Namitg02 commited on
Commit
b84e0a2
·
verified ·
1 Parent(s): 401b243

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def talk(prompt, history):
144
  ]
145
  # indicates the end of a sequence
146
 
147
- output = model.create_chat_completion(messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}], max_tokens=1000, stop=["</s>"], stream=True)
148
  print(output['choices'][0]['message']['content'])
149
 
150
  # for output in stream:
 
144
  ]
145
  # indicates the end of a sequence
146
 
147
+ output = model.create_chat_completion(messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}], max_tokens=1000, stop=["</s>"], stream=False)
148
  print(output['choices'][0]['message']['content'])
149
 
150
  # for output in stream: