Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -17,11 +17,6 @@ for gpu in gpus:
|
|
17 |
tf.config.experimental.set_memory_growth(gpu, True)
|
18 |
except RuntimeError as e:
|
19 |
print(f"[TF] GPU memory growth config error: {e}")
|
20 |
-
|
21 |
-
# Limit CPU memory usage (for Hugging Face Spaces)
|
22 |
-
tf.config.set_logical_device_configuration(
|
23 |
-
tf.config.list_physical_devices('CPU')[0],
|
24 |
-
[tf.config.LogicalDeviceConfiguration(memory_limit=2048)]
|
25 |
)
|
26 |
# Initialize AI Core for TB analysis
|
27 |
ai_core = AICoreAGIX()
|
|
|
17 |
tf.config.experimental.set_memory_growth(gpu, True)
|
18 |
except RuntimeError as e:
|
19 |
print(f"[TF] GPU memory growth config error: {e}")
|
|
|
|
|
|
|
|
|
|
|
20 |
)
|
21 |
# Initialize AI Core for TB analysis
|
22 |
ai_core = AICoreAGIX()
|