Spaces:
Running
Running
Correct model paths
Browse files- gradio_utils.py +4 -4
gradio_utils.py
CHANGED
|
@@ -64,13 +64,13 @@ def get_video_from_canny_selection(canny_selection):
|
|
| 64 |
|
| 65 |
def get_model_from_db_selection(db_selection):
|
| 66 |
if db_selection == "Anime DB":
|
| 67 |
-
input_video_path = 'PAIR/controlnet-canny-anime'
|
| 68 |
elif db_selection == "Avatar DB":
|
| 69 |
-
input_video_path = 'PAIR/controlnet-canny-avatar'
|
| 70 |
elif db_selection == "GTA-5 DB":
|
| 71 |
-
input_video_path = 'PAIR/controlnet-canny-gta5'
|
| 72 |
elif db_selection == "Arcane DB":
|
| 73 |
-
input_video_path = 'PAIR/controlnet-canny-arcane'
|
| 74 |
else:
|
| 75 |
input_video_path = db_selection
|
| 76 |
|
|
|
|
| 64 |
|
| 65 |
def get_model_from_db_selection(db_selection):
|
| 66 |
if db_selection == "Anime DB":
|
| 67 |
+
input_video_path = 'PAIR/text2video-zero-controlnet-canny-anime'
|
| 68 |
elif db_selection == "Avatar DB":
|
| 69 |
+
input_video_path = 'PAIR/text2video-zero-controlnet-canny-avatar'
|
| 70 |
elif db_selection == "GTA-5 DB":
|
| 71 |
+
input_video_path = 'PAIR/text2video-zero-controlnet-canny-gta5'
|
| 72 |
elif db_selection == "Arcane DB":
|
| 73 |
+
input_video_path = 'PAIR/text2video-zero-controlnet-canny-arcane'
|
| 74 |
else:
|
| 75 |
input_video_path = db_selection
|
| 76 |
|