awacke1 commited on
Commit
32d6789
·
1 Parent(s): 803e57f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -204,13 +204,13 @@ def chat_with_model(prompt, document_section, model_choice='Llama-2-7b-chat-hf')
204
  collected_chunks.append(r.token.text)
205
  chunk_message = r.token.text
206
  collected_messages.append(chunk_message)
207
- #try:
208
  report.append(r.token.text)
209
  if len(r.token.text) > 0:
210
  result="".join(report).strip()
211
  res_box.markdown(f'*{result}*')
212
 
213
- #except:
214
  st.write('Stream llm issue')
215
  st.write("Elapsed time:")
216
  st.write(time.time() - start_time)
 
204
  collected_chunks.append(r.token.text)
205
  chunk_message = r.token.text
206
  collected_messages.append(chunk_message)
207
+ try:
208
  report.append(r.token.text)
209
  if len(r.token.text) > 0:
210
  result="".join(report).strip()
211
  res_box.markdown(f'*{result}*')
212
 
213
+ except:
214
  st.write('Stream llm issue')
215
  st.write("Elapsed time:")
216
  st.write(time.time() - start_time)