Update app.py
Browse files
app.py
CHANGED
@@ -152,7 +152,7 @@ def talk(prompt, history):
|
|
152 |
# indicates the end of a sequence
|
153 |
text = ""
|
154 |
stream = model.create_chat_completion(messages, max_tokens=1000, stop=["</s>"], stream=True)
|
155 |
-
print(
|
156 |
# for output in stream:
|
157 |
# text += output["choices"][0]["text"]
|
158 |
# yield text
|
|
|
152 |
# indicates the end of a sequence
|
153 |
text = ""
|
154 |
stream = model.create_chat_completion(messages, max_tokens=1000, stop=["</s>"], stream=True)
|
155 |
+
print(stream["choices"][0]["text"])
|
156 |
# for output in stream:
|
157 |
# text += output["choices"][0]["text"]
|
158 |
# yield text
|