nianlonggu commited on
Commit
8589755
1 Parent(s): a877983
Files changed (1) hide show
  1. services/app.py +4 -0
services/app.py CHANGED
@@ -75,6 +75,8 @@ def main():
75
  paper_info = convert_pdf_to_json( pdf_bytes )
76
  extracted_summary = summarize_paper(memsum, paper_info)
77
  st.markdown( "\n".join( [ "* " + sen for sen in extracted_summary] ) )
 
 
78
 
79
  elif option == 'Directly upload a PDF':
80
  # User can upload a PDF directly here
@@ -88,6 +90,8 @@ def main():
88
  paper_info = convert_pdf_to_json( pdf_bytes )
89
  extracted_summary = summarize_paper(memsum, paper_info)
90
  st.markdown( "\n".join( [ "* " + sen for sen in extracted_summary] ) )
 
 
91
 
92
 
93
  if __name__ == "__main__":
 
75
  paper_info = convert_pdf_to_json( pdf_bytes )
76
  extracted_summary = summarize_paper(memsum, paper_info)
77
  st.markdown( "\n".join( [ "* " + sen for sen in extracted_summary] ) )
78
+ else:
79
+ st.markdown("(This may takes upto 1 min on huggingface space, since MemSum is running on a small CPU)")
80
 
81
  elif option == 'Directly upload a PDF':
82
  # User can upload a PDF directly here
 
90
  paper_info = convert_pdf_to_json( pdf_bytes )
91
  extracted_summary = summarize_paper(memsum, paper_info)
92
  st.markdown( "\n".join( [ "* " + sen for sen in extracted_summary] ) )
93
+ else:
94
+ st.markdown("(This may takes upto 1 min on huggingface space, since MemSum is running on a small CPU)")
95
 
96
 
97
  if __name__ == "__main__":