wangrongsheng commited on
Commit
c90545b
·
1 Parent(s): 046f527

support two models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ def generate(
55
  top_k: int = 50,
56
  repetition_penalty: float = 1.2,
57
  ) -> Iterator[str]:
58
- if select_model=="INTJ-en"
59
  conversation = []
60
  if system_prompt:
61
  conversation.append({"role": "system", "content": system_prompt})
@@ -89,7 +89,7 @@ def generate(
89
  outputs.append(text)
90
  yield "".join(outputs)
91
 
92
- if select_model=="INTJ-zh"
93
  conversation = []
94
  if system_prompt:
95
  conversation.append({"role": "system", "content": system_prompt})
 
55
  top_k: int = 50,
56
  repetition_penalty: float = 1.2,
57
  ) -> Iterator[str]:
58
+ if select_model=="INTJ-en":
59
  conversation = []
60
  if system_prompt:
61
  conversation.append({"role": "system", "content": system_prompt})
 
89
  outputs.append(text)
90
  yield "".join(outputs)
91
 
92
+ if select_model=="INTJ-zh":
93
  conversation = []
94
  if system_prompt:
95
  conversation.append({"role": "system", "content": system_prompt})