Namitg02 commited on
Commit
31d0c60
·
verified ·
1 Parent(s): 1ccd79e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -142,7 +142,8 @@ def talk(prompt, history):
142
  print("check6")
143
 
144
  # indicates the end of a sequence
145
- stream = model.create_chat_completion(messages = messages, max_tokens=1000, stop=["</s>"], stream=False)
 
146
  print(f"{stream}")
147
  print("check 7")
148
  print(stream['choices'][0]['message']['content'])
 
142
  print("check6")
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(messages = 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'])