Spaces:

Vchitect
/

VBench_Video_Arena

Running

ynhe commited on Jan 7

Commit

e8989f9

verified ·

1 Parent(s): cf8d75e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,8 +61,8 @@ def get_video_path_local(model_name, type, prompt):
     else:
         video_path_subfolder = os.path.join(model_name, type)
-    if 'cogvideo' in model_name:
-        prompt.replace(".mp4",".gif")
     try:
         return hf_api.hf_hub_download(

     else:
         video_path_subfolder = os.path.join(model_name, type)
+    if model_name == 'cogvideo':
+        prompt = prompt.replace(".mp4",".gif")
     try:
         return hf_api.hf_hub_download(