Spaces:
Sleeping
Sleeping
Update clothGen.py
Browse files- clothGen.py +3 -2
clothGen.py
CHANGED
@@ -11,7 +11,8 @@ na_prompt = na_prompt_file.to_string(index=False)
|
|
11 |
|
12 |
|
13 |
def cloth_gen(advice, gender):
|
14 |
-
|
|
|
15 |
if gender == "男":
|
16 |
lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NA.safetensors"
|
17 |
else:
|
@@ -23,7 +24,7 @@ def cloth_gen(advice, gender):
|
|
23 |
cloth_intro = ""
|
24 |
promptGen = ""
|
25 |
if start_index != -1:
|
26 |
-
start_index += len("prompt\n
|
27 |
end_index = prompt.find("promptEnd")
|
28 |
if end_index != -1:
|
29 |
extracted_content = prompt[start_index:end_index]
|
|
|
11 |
|
12 |
|
13 |
def cloth_gen(advice, gender):
|
14 |
+
print(f"cloth_gen_gender{gender}")
|
15 |
+
# lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NV.safetensors"
|
16 |
if gender == "男":
|
17 |
lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NA.safetensors"
|
18 |
else:
|
|
|
24 |
cloth_intro = ""
|
25 |
promptGen = ""
|
26 |
if start_index != -1:
|
27 |
+
start_index += len("prompt\n")
|
28 |
end_index = prompt.find("promptEnd")
|
29 |
if end_index != -1:
|
30 |
extracted_content = prompt[start_index:end_index]
|