Update app.py

app.py

@@ -5,6 +5,14 @@ import asyncio
 sys.path.append("/home/user/app/components")
 from HuggingFaceHelper import HuggingFaceHelper
 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)