Update app.py

app.py

@@ -2,20 +2,35 @@ import torch
 import gradio as gr
 import imageio
 import os
+import requests
 from safetensors.torch import load_file
 from torchvision import transforms
 from PIL import Image
 import numpy as np
 
-# Define model path (assuming it's in the HF Space)
-MODEL_PATH = "sarthak247/Wan2.1-T2V-1.3B-nf4"
-MODEL_FILE = f"{MODEL_PATH}/blob/main/diffusion_pytorch_model.safetensors"
+# Define model URL and local path
+MODEL_URL = "https://huggingface.co/sarthak247/Wan2.1-T2V-1.3B-nf4/resolve/main/diffusion_pytorch_model.safetensors"
+MODEL_FILE = "diffusion_pytorch_model.safetensors"
+
+# Function to download model if not present
+def download_model():
+    if not os.path.exists(MODEL_FILE):
+        print("Downloading model...")
+        response = requests.get(MODEL_URL, stream=True)
+        if response.status_code == 200:
+            with open(MODEL_FILE, "wb") as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            print("Download complete!")
+        else:
+            raise RuntimeError(f"Failed to download model: {response.status_code}")
 
 # Load model weights manually
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Loading model on {device}...")
 
 try:
+    download_model()
     model_weights = load_file(MODEL_FILE, device=device)
     print("Model loaded successfully!")
 except Exception as e: