Update app.py

app.py

@@ -5,7 +5,18 @@ import asyncio
 sys.path.append("/home/user/app/components")
 from HuggingFaceHelper import HuggingFaceHelper
 from AICoreAGIX_with_TB import AICoreAGIX
+import tensorflow as tf
 
+# Limit GPU memory usage (important for low-memory environments)
+gpus = tf.config.experimental.list_physical_devices('GPU')
+for gpu in gpus:
+    tf.config.experimental.set_memory_growth(gpu, True)
+
+# Limit CPU memory usage (for Hugging Face Spaces)
+tf.config.set_logical_device_configuration(
+    tf.config.list_physical_devices('CPU')[0],
+    [tf.config.LogicalDeviceConfiguration(memory_limit=2048)]
+)
 # Initialize AI Core for TB analysis
 ai_core = AICoreAGIX()