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