Update app.py

app.py

@@ -10,6 +10,13 @@ import os
 from huggingface_hub import login
 import os
 
+class MemoryTracker:
+    @staticmethod
+    def get_memory_usage():
+        process = psutil.Process(os.getpid())
+        memory_gb = process.memory_info().rss / 1024 / 1024 / 1024
+        return f"{memory_gb:.2f} GB"
+
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)