Spaces:
Running
Running
Update app.py
Browse files
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,
|