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