thigobr commited on
Commit
c552ef8
·
verified ·
1 Parent(s): 209a678

Print stats

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -35,6 +35,10 @@ def main():
35
  vector_store = PDF_4_QA(uploaded_file)
36
  QA_Bot(vector_store)
37
  profiler.disable()
 
 
 
 
38
 
39
  if __name__ == '__main__':
40
  main()
 
35
  vector_store = PDF_4_QA(uploaded_file)
36
  QA_Bot(vector_store)
37
  profiler.disable()
38
+ stats = pstats.Stats(profiler).sort_stats('cumulative')
39
+ #stats.print_stats()
40
+ st.session_state.messages.append({"role": "assistant", "content": stats})
41
+
42
 
43
  if __name__ == '__main__':
44
  main()