yupeng.zhou commited on
Commit
a33dac7
1 Parent(s): 01c86f9
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -489,8 +489,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
489
  raise gr.Error("Please add the triger word \" img \" behind the class word you want to customize, such as: man img or woman img")
490
  if _upload_images is None and _model_type != "original":
491
  raise gr.Error(f"Cannot find any input face image!")
492
- if len(prompt_array) > 10:
493
- print(prompt_array)
494
  raise gr.Error(f"No more than 10 prompts in huggface demo for Speed! But found {len(prompt_array)} prompts!")
495
  global sa32, sa64,id_length,total_length,attn_procs,unet,cur_model_type,device
496
  global write
 
489
  raise gr.Error("Please add the triger word \" img \" behind the class word you want to customize, such as: man img or woman img")
490
  if _upload_images is None and _model_type != "original":
491
  raise gr.Error(f"Cannot find any input face image!")
492
+ if len(prompt_array.splitlines()) > 10:
 
493
  raise gr.Error(f"No more than 10 prompts in huggface demo for Speed! But found {len(prompt_array)} prompts!")
494
  global sa32, sa64,id_length,total_length,attn_procs,unet,cur_model_type,device
495
  global write