Update app.py

app.py

@@ -59,11 +59,11 @@ def get_random_video():
 
     video_path_subfolder = os.path.join(model_name, type)
     try:
-        hf_api.hf_hub_download(
+        return hf_api.hf_hub_download(
             repo_id = repo_id,
             filename = prompt,
             subfolder = video_path_subfolder,
-            repo_type = dataset,
+            repo_type = "dataset",
             local_dir = local_dir
         )
     except Exception as e:
@@ -71,18 +71,18 @@ def get_random_video():
         print(e)
         video_path_subfolder = model_name
         try:
-            hf_api.hf_hub_download(
+            return hf_api.hf_hub_download(
                 repo_id = repo_id,
                 filename = prompt,
                 subfolder = video_path_subfolder,
-                repo_type = dataset,
+                repo_type = 'dataset',
                 local_dir = local_dir
             )
         except Exception as e:
             print(e)
     # video_path = dataset['train'][random_index]['video_path']
-    print('error:', video_path)
-    return video_path
+    print('error:', mdoel_name, type, prompt)
+    return type
 
 # Gradio 接口
 def display_video():