Update app.py

app.py

@@ -524,8 +524,19 @@ class ModelWrapper:
 # GRADIO INTERFACE
 # ==============================================================================
 
+# Download and load model from HuggingFace Hub
+from huggingface_hub import snapshot_download
+
+print("📥 Downloading model from HuggingFace Hub...")
+model_path = snapshot_download(
+    repo_id="Smilyai-labs/MixSam-exp",
+    repo_type="model",
+    local_dir="./model_cache"
+)
+print(f"✅ Model downloaded to: {model_path}")
+
 # Load model
-model = ModelWrapper("Smilyai-labs/MixSam-exp")
+model = ModelWrapper(model_path)
 
 def chat_fn(message, history, temperature, top_k, top_p, max_tokens):
     # Build conversation context