Update app.py
Browse files
app.py
CHANGED
|
@@ -143,7 +143,7 @@ def talk(prompt, history):
|
|
| 143 |
|
| 144 |
# indicates the end of a sequence
|
| 145 |
# stream = model.create_chat_completion(messages = messages, max_tokens=1000, stop=["</s>"], stream=False)
|
| 146 |
-
stream = model(
|
| 147 |
print(f"{stream}")
|
| 148 |
print("check 7")
|
| 149 |
print(stream['choices'][0]['message']['content'])
|
|
|
|
| 143 |
|
| 144 |
# indicates the end of a sequence
|
| 145 |
# stream = model.create_chat_completion(messages = messages, max_tokens=1000, stop=["</s>"], stream=False)
|
| 146 |
+
stream = model.create_completion(prompt = messages, max_tokens=1000, stop=["</s>"],echo=True, stream=False)
|
| 147 |
print(f"{stream}")
|
| 148 |
print("check 7")
|
| 149 |
print(stream['choices'][0]['message']['content'])
|