Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,11 +2,11 @@ import streamlit as st
|
|
| 2 |
from transformers import pipeline
|
| 3 |
|
| 4 |
@st.cache(allow_output_mutation=True)
|
| 5 |
-
|
| 6 |
-
summarizer = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
|
| 7 |
-
|
| 8 |
|
| 9 |
-
|
| 10 |
st.title("Patent Text Summarizer")
|
| 11 |
sentence = st.text_area('Please paste your Patent Text :', height=300)
|
| 12 |
button = st.button("Summarize")
|
|
@@ -17,7 +17,7 @@ button = st.button("Summarize")
|
|
| 17 |
with st.spinner("Generating Patent Summary.."):
|
| 18 |
if button and sentence:
|
| 19 |
#chunks = generate_chunks(sentence)
|
| 20 |
-
res =
|
| 21 |
max_length=500,
|
| 22 |
min_length=100,
|
| 23 |
truncation=True,
|
|
|
|
| 2 |
from transformers import pipeline
|
| 3 |
|
| 4 |
@st.cache(allow_output_mutation=True)
|
| 5 |
+
def load_summarizer():
|
| 6 |
+
summarizer = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
|
| 7 |
+
return summarizer
|
| 8 |
|
| 9 |
+
summarize = load_summarizer()
|
| 10 |
st.title("Patent Text Summarizer")
|
| 11 |
sentence = st.text_area('Please paste your Patent Text :', height=300)
|
| 12 |
button = st.button("Summarize")
|
|
|
|
| 17 |
with st.spinner("Generating Patent Summary.."):
|
| 18 |
if button and sentence:
|
| 19 |
#chunks = generate_chunks(sentence)
|
| 20 |
+
res = summarize(sentence,
|
| 21 |
max_length=500,
|
| 22 |
min_length=100,
|
| 23 |
truncation=True,
|