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})
|