Spaces:
Runtime error
Runtime error
Commit
·
c90545b
1
Parent(s):
046f527
support two models
Browse files
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})
|