ynhe commited on
Commit
66fcddf
·
verified ·
1 Parent(s): 2dff84c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -60,6 +60,10 @@ def get_video_path_local(model_name, type, prompt):
60
  video_path_subfolder = os.path.join(model_name, type, '1024x576')
61
  else:
62
  video_path_subfolder = os.path.join(model_name, type)
 
 
 
 
63
  try:
64
  return hf_api.hf_hub_download(
65
  repo_id = repo_id,
 
60
  video_path_subfolder = os.path.join(model_name, type, '1024x576')
61
  else:
62
  video_path_subfolder = os.path.join(model_name, type)
63
+
64
+ if model_name=='cogvideo':
65
+ prompt.replace(".mp4",".gif")
66
+
67
  try:
68
  return hf_api.hf_hub_download(
69
  repo_id = repo_id,