frankai98 commited on
Commit
03d0510
·
verified ·
1 Parent(s): 996f66f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -294,7 +294,7 @@ def main():
294
  st.error(f"Gemma processing failed: {str(error)}")
295
  report = "Error generating report. Please try again with fewer tweets."
296
  else:
297
- report = raw_result
298
  # report = extract_assistant_response(raw_result)
299
  clear_gpu_memory()
300
  progress_bar.progress(100)
@@ -302,10 +302,10 @@ def main():
302
  html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
303
  st.session_state.timer_frozen = True
304
 
305
- # formatted_report = report.replace('\n', '<br>')
306
 
307
  st.subheader("Generated Report:")
308
- st.markdown(f"<div style='font-size: normal; font-weight: normal;'>{report}</div>", unsafe_allow_html=True)
309
 
310
  if __name__ == '__main__':
311
  main()
 
294
  st.error(f"Gemma processing failed: {str(error)}")
295
  report = "Error generating report. Please try again with fewer tweets."
296
  else:
297
+ report = raw_result[0][0]['generated_text']
298
  # report = extract_assistant_response(raw_result)
299
  clear_gpu_memory()
300
  progress_bar.progress(100)
 
302
  html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
303
  st.session_state.timer_frozen = True
304
 
305
+ formatted_report = report.replace('\n', '<br>')
306
 
307
  st.subheader("Generated Report:")
308
+ st.markdown(f"<div style='font-size: normal; font-weight: normal;'>{formatted_report}</div>", unsafe_allow_html=True)
309
 
310
  if __name__ == '__main__':
311
  main()