Namitg02 commited on
Commit
311ac6d
·
verified ·
1 Parent(s): f0cb2fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -151,7 +151,7 @@ def talk(prompt, history):
151
  ]
152
  # indicates the end of a sequence
153
  text = ""
154
- stream = model(messages, max_tokens=1000, stop=["</s>"], stream=True)
155
  for output in stream:
156
  text += output["choices"][0]["text"]
157
  yield text
 
151
  ]
152
  # indicates the end of a sequence
153
  text = ""
154
+ stream = model.create_chat_completion(messages, max_tokens=1000, stop=["</s>"], stream=True)
155
  for output in stream:
156
  text += output["choices"][0]["text"]
157
  yield text