jiayong commited on
Commit
48d6594
1 Parent(s): a20d9f0

Update gen_client.py

Browse files
Files changed (1) hide show
  1. gen_client.py +1 -4
gen_client.py CHANGED
@@ -451,12 +451,9 @@ class HumanGenService:
451
  if file_name not in template_prompts:
452
  raise gr.Error("The input video is not a template!")
453
  input_prompt = template_prompts[file_name]
454
- if file_name not in template_orign_videos:
455
- raise gr.Error("The input video is not a template!")
456
- ref_ori_video_path = template_orign_videos[file_name]
457
  else:
458
  print("video file not found:%s" % ref_video_name)
459
- return ref_ori_video_path, input_prompt
460
 
461
  def click_button_func_async(self, user_id, request_id, input_mode, ref_image_path, ref_video_path, input_prompt='', prompt_template='',model_id=False):
462
  start_time = time.time()
 
451
  if file_name not in template_prompts:
452
  raise gr.Error("The input video is not a template!")
453
  input_prompt = template_prompts[file_name]
 
 
 
454
  else:
455
  print("video file not found:%s" % ref_video_name)
456
+ return input_prompt
457
 
458
  def click_button_func_async(self, user_id, request_id, input_mode, ref_image_path, ref_video_path, input_prompt='', prompt_template='',model_id=False):
459
  start_time = time.time()