Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def talk(prompt, history):
|
|
126 |
# the chat template structure should be based on text generation model format
|
127 |
|
128 |
# indicates the end of a sequence
|
129 |
-
stream = model.create_chat_completion(messages = messages, stop=["</s>"], stream=False)
|
130 |
# print(f"{stream}")
|
131 |
print("check 7")
|
132 |
global historylog
|
|
|
126 |
# the chat template structure should be based on text generation model format
|
127 |
|
128 |
# indicates the end of a sequence
|
129 |
+
stream = model.create_chat_completion(messages = messages,max_tokens =400, stop=["</s>"], stream=False)
|
130 |
# print(f"{stream}")
|
131 |
print("check 7")
|
132 |
global historylog
|