Hugging Face's logo

Spaces:
trhacknon
/
Text2Video-ZeroTrkn
Runtime error

App Files Community
lev1 commited on
Commit
9adc565
·
1 Parent(s): 2751474

Correct model paths

Browse files
Files changed (1) hide show
  1. 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