SIUBIU commited on
Commit
0128016
·
verified ·
1 Parent(s): 8d805f4

Update clothGen.py

Browse files
Files changed (1) hide show
  1. 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
- lora_path = "https://huggingface.co/PPSharks/PPSharksModels/resolve/main/NV.safetensors"
 
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\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]