wjjessen commited on
Commit
052edd8
·
1 Parent(s): 7edcbdb

added copy summary text

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -5,10 +5,12 @@ from langchain.docstore.document import Document
5
  from langchain.document_loaders.pdf import PyMuPDFLoader
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from PyPDF2 import PdfReader
 
8
  import streamlit as st
 
 
9
  import torch
10
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
11
- import time
12
 
13
  # notes
14
  # https://huggingface.co/docs/transformers/pad_truncation
@@ -181,8 +183,12 @@ def main():
181
  + "  |  Summarization time: "
182
  f"{duration:.0f}" + " seconds"
183
  )
184
- st.success(summary)
185
-
 
 
 
 
186
 
187
  st.markdown(
188
  """<style>
 
5
  from langchain.document_loaders.pdf import PyMuPDFLoader
6
  from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from PyPDF2 import PdfReader
8
+ import pyperclip
9
  import streamlit as st
10
+ import textwrap as tw
11
+ import time
12
  import torch
13
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
 
14
 
15
  # notes
16
  # https://huggingface.co/docs/transformers/pad_truncation
 
183
  + "&nbsp;&nbsp;|&nbsp;&nbsp;Summarization time: "
184
  f"{duration:.0f}" + " seconds"
185
  )
186
+ st.code(
187
+ "\n".join(
188
+ tw.wrap(
189
+ summary,
190
+ width=80)), language='md')
191
+
192
 
193
  st.markdown(
194
  """<style>