Update app.py
Browse files
app.py
CHANGED
|
@@ -144,7 +144,7 @@ def talk(prompt, history):
|
|
| 144 |
]
|
| 145 |
# indicates the end of a sequence
|
| 146 |
|
| 147 |
-
output = model.create_chat_completion(messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}], max_tokens=1000, stop=["</s>"], stream=
|
| 148 |
print(output['choices'][0]['message']['content'])
|
| 149 |
|
| 150 |
# for output in stream:
|
|
|
|
| 144 |
]
|
| 145 |
# indicates the end of a sequence
|
| 146 |
|
| 147 |
+
output = model.create_chat_completion(messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}], max_tokens=1000, stop=["</s>"], stream=False)
|
| 148 |
print(output['choices'][0]['message']['content'])
|
| 149 |
|
| 150 |
# for output in stream:
|