Update app.py

app.py

@@ -1,11 +1,22 @@
 import gradio as gr
 from modelscope.pipelines import pipeline
 from modelscope.outputs import OutputKeys
+from huggingface_hub import snapshot_download, HfApi
+import pathlib
 
-p = pipeline('text-to-video-synthesis', 'damo/text-to-video-synthesis')
+HUB_TOKEN = os.getenv("HUB_TOKEN")
+api = HfApi()
+api.set_access_token(HUB_TOKEN)
+model_dir = pathlib.Path('weights')
+snapshot_download('damo-vilab/modelscope-damo-text-to-video-synthesis',
+                  repo_type='model', 
+                  local_dir=model_dir, 
+                  use_auth_token=True)
 
-def generate_video(text_input):
-    output_video_path = p({'text': text_input})[OutputKeys.OUTPUT_VIDEO]
+pipe = pipeline('text-to-video-synthesis', model_dir.as_posix())
+
+def generate_video(prompt):
+    output_video_path = pipe({'text': prompt})[OutputKeys.OUTPUT_VIDEO]
     return output_video_path
 
 iface = gr.Interface(