Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 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 |
-
|
| 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)
|