CLOTH_WEB_V1_simple

Sleeping

SIUBIU commited on Oct 15

Commit

a5fe8d0

•

1 Parent(s): af149d1

Update prompt_gen.py

Files changed (1) hide show

prompt_gen.py CHANGED Viewed

@@ -12,8 +12,8 @@ os.environ["OPENAI_BASE_URL"] = "http://15.204.101.64:4000/v1"
 def prompt_gen(advise, gender):
-    prompt = na_prompt
-    trigger = "A Hanfu"
     if gender == "男":
         prompt = na_prompt
         trigger = "A Hanfu"

 def prompt_gen(advise, gender):
+    prompt = nv_prompt
+    trigger = "a Hanfu"
     if gender == "男":
         prompt = na_prompt
         trigger = "A Hanfu"