Update app.py

app.py

@@ -13,15 +13,9 @@ from diffsynth import ModelManager, WanVideoPipeline, save_video
 
 num_frames, width, height = 49, 832, 480
 gpu_id = 0
-device = f'cuda:{gpu_id}'
-
-from huggingface_hub import login
-import os
-# 使用环境变量中的 token 登录
-token = os.environ.get("hf_sapce_token")
-if token:
-    login(token)
+device = f'cuda:{gpu_id}' if torch.cuda.is_available() else 'cpu'
 
+os.system("wget https://huggingface.co/alibaba-pai/Wan2.1-Fun-1.3B-Control/resolve/main/Wan2.1_VAE.pth")
 
 rmbg_model = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
 torch.set_float32_matmul_precision(['high', 'highest'][0])