Namitg02 commited on
Commit
f2b57d7
·
verified ·
1 Parent(s): 7b9f78b

Update app.py

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