Update app.py

app.py

@@ -51,6 +51,7 @@ def get_random_video():
     # 随机选择一个Prompt
     random_index = random.randint(0, len(dimension[type]) - 1)
     prompt = dimension[type][random_index]
+    prompt = os.path.basename(prompt)
     # 随机一个模型
     random_index = random.randint(0, len(model_names) - 1)
     model_name =  model_names[random_index]
@@ -67,8 +68,7 @@ def get_random_video():
             local_dir = local_dir
         )
     except Exception as e:
-        print(f"[PATH]{video_path_subfolder} NOT in hf repo, try {model_name}")
-        print(e)
+        print(f"[PATH]{video_path_subfolder}/{prompt} NOT in hf repo, try {model_name}",e)
         video_path_subfolder = model_name
         try:
             return hf_api.hf_hub_download(
@@ -79,9 +79,10 @@ def get_random_video():
                 local_dir = local_dir
             )
         except Exception as e:
+            print(f"[PATH]{video_path_subfolder}/{prompt} NOT in hf repo, try {model_name}",e)
             print(e)
     # video_path = dataset['train'][random_index]['video_path']
-    print('error:', mdoel_name, type, prompt)
+    print('error:', model_name, type, prompt)
     return type
 
 # Gradio 接口