Update app.py

app.py

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