Spaces:
Running
Running
Update clothGen.py
Browse files- 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("
|
|
|
|
|
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 = ""
|