Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
 
-# 加載模型
-pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+# 使用另一個更穩定的模型
+pipe = DiffusionPipeline.from_pretrained("ali-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16)
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
 # 定義生成視頻的函數
 def text_to_video(prompt):