hf chnages new

Dockerfile

@@ -22,17 +22,8 @@ COPY backend/ ./backend/
 COPY --from=frontend-builder /frontend/dist ./frontend/dist
 
 # Pre-download model so container starts fast (avoids HF Spaces startup timeout)
-RUN python -c "
-import os
-os.makedirs('models', exist_ok=True)
-from huggingface_hub import hf_hub_download
-hf_hub_download(
-    repo_id='bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF',
-    filename='Qwen2.5-Coder-0.5B-Instruct-Q4_K_M.gguf',
-    local_dir='models',
-    local_dir_use_symlinks=False
-)
-"
+COPY scripts/setup_model.py /tmp/setup_model.py
+RUN python /tmp/setup_model.py && rm /tmp/setup_model.py
 
 ENV PORT=8000 HOST=0.0.0.0 PYTHONPATH=/app
 

scripts/setup_model.py

@@ -0,0 +1,28 @@
+import os
+import sys
+from huggingface_hub import hf_hub_download
+
+def download_model():
+    """Download the required model for the application."""
+    try:
+        print("Creating models directory...")
+        os.makedirs('models', exist_ok=True)
+        
+        print("Downloading model from Hugging Face...")
+        hf_hub_download(
+            repo_id='bartowski/Qwen2.5-Coder-0.5B-Instruct-GGUF',
+            filename='Qwen2.5-Coder-0.5B-Instruct-Q4_K_M.gguf',
+            local_dir='models',
+            local_dir_use_symlinks=False
+        )
+        
+        print("Model downloaded successfully!")
+        return True
+        
+    except Exception as e:
+        print(f"Error downloading model: {e}", file=sys.stderr)
+        return False
+
+if __name__ == "__main__":
+    success = download_model()
+    sys.exit(0 if success else 1)