Update app.py

app.py

@@ -1,15 +1,17 @@
-import torch
-from diffusers import AutoencoderKLWan, WanImageToVideoPipeline, UniPCMultistepScheduler
-from diffusers.utils import export_to_video
-from transformers import CLIPVisionModel
+import spaces  # MUST be imported FIRST before any torch/CUDA imports
 import gradio as gr
 import tempfile
-import spaces
 from huggingface_hub import hf_hub_download
 import numpy as np
 from PIL import Image
 import random
 
+# Now import torch and related modules AFTER spaces
+import torch
+from diffusers import AutoencoderKLWan, WanImageToVideoPipeline, UniPCMultistepScheduler
+from diffusers.utils import export_to_video
+from transformers import CLIPVisionModel
+
 MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 LORA_REPO_ID = "Kijai/WanVideo_comfy"
 LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"