added copy summary text
Browse files
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.
|
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 |
+ " | 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>
|