nianlonggu
commited on
Commit
•
8589755
1
Parent(s):
a877983
update
Browse files- 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__":
|