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,
|