SIUBIU commited on
Commit
aff69d1
·
verified ·
1 Parent(s): 7e0ad3e

Update prompt_gen.py

Browse files
Files changed (1) hide show
  1. prompt_gen.py +3 -2
prompt_gen.py CHANGED
@@ -12,11 +12,12 @@ os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1"
12
 
13
 
14
  def prompt_gen(advise, gender):
15
- print(gender)
 
16
  if gender == "男":
17
  prompt = na_prompt
18
  trigger = "A Hanfu"
19
- else:
20
  prompt = nv_prompt
21
  trigger = "a Hanfu"
22
 
 
12
 
13
 
14
  def prompt_gen(advise, gender):
15
+ prompt = na_prompt
16
+ trigger = "A Hanfu"
17
  if gender == "男":
18
  prompt = na_prompt
19
  trigger = "A Hanfu"
20
+ else if gender == "女":
21
  prompt = nv_prompt
22
  trigger = "a Hanfu"
23