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

Update app.py

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