Update app.py

app.py

@@ -21,6 +21,7 @@ import hashlib
 from datetime import datetime
 from typing import Dict, List, Optional
 from huggingface_hub import login, HfApi
+from types import SimpleNamespace
 
 # ---------------------- UTILITY FUNCTIONS ----------------------
 
@@ -362,6 +363,10 @@ def estimate_memory_requirements(model_path, precision):
         # Calculate total required memory
         required_memory = (base_memory + (model_size if model_size else 12)) * memory_multiplier
         
+        # Adjust for Hugging Face Space
+        if required_memory > 16:
+            required_memory = 16  # Cap at 16GB for Hugging Face Space
+        
         return required_memory
     except:
         return 16  # Default safe estimate