SIUBIU commited on
Commit
4387f4d
1 Parent(s): a5fe8d0

Update clothGen.py

Browse files
Files changed (1) hide show
  1. clothGen.py +3 -1
clothGen.py CHANGED
@@ -18,7 +18,9 @@ def cloth_gen(advice, gender):
18
  lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NV.safetensors"
19
 
20
  prompt = prompt_gen(advice, gender)
21
- start_index = prompt.find("prompt" or "Prompt")
 
 
22
  intro_index = prompt.find("服饰风格介绍")
23
  cloth_intro = ""
24
  promptGen = ""
 
18
  lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NV.safetensors"
19
 
20
  prompt = prompt_gen(advice, gender)
21
+ start_index = prompt.find("Prompt")
22
+ if start_index == -1:
23
+ start_index = prompt.find("prompt")
24
  intro_index = prompt.find("服饰风格介绍")
25
  cloth_intro = ""
26
  promptGen = ""