Update app.py

app.py

@@ -8,17 +8,15 @@ from AICoreAGIX_with_TB import AICoreAGIX
 import os
 os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
 
-import tensorflow as tf
-tf.config.set_logical_device_configuration(
-    tf.config.list_physical_devices('CPU')[0],
-    [tf.config.LogicalDeviceConfiguration(memory_limit=2048)]
-)
 import tensorflow as tf
 
-# Limit GPU memory usage (important for low-memory environments)
+# Limit GPU memory usage (if GPU exists)
 gpus = tf.config.experimental.list_physical_devices('GPU')
 for gpu in gpus:
-    tf.config.experimental.set_memory_growth(gpu, True)
+    try:
+        tf.config.experimental.set_memory_growth(gpu, True)
+    except RuntimeError as e:
+        print(f"[TF] GPU memory growth config error: {e}")
 
 # Limit CPU memory usage (for Hugging Face Spaces)
 tf.config.set_logical_device_configuration(