Namitg02 commited on
Commit
84ef949
·
verified ·
1 Parent(s): 900c9eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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