MrGanesh commited on
Commit
13fcde7
1 Parent(s): 2c95677

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
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
- def load_summarizer():
6
- model = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
7
- return model
8
 
9
- summarizer = 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,9 +17,10 @@ button = st.button("Summarize")
17
  with st.spinner("Generating Patent Summary.."):
18
  if button and sentence:
19
  #chunks = generate_chunks(sentence)
20
- res = summarizer(sentence,
21
  max_length=500,
22
- min_length=100,
 
23
  #do_sample=do_sample
24
  )
25
  text = ' '.join([summ['summary_text'] for summ in res])
 
2
  from transformers import pipeline
3
 
4
  @st.cache(allow_output_mutation=True)
5
+ #def load_summarizer():
6
+ model = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
7
+ #return model
8
 
9
+ #summarizer = 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 = model(sentence,
21
  max_length=500,
22
+ min_length=100,
23
+ truncation=True,
24
  #do_sample=do_sample
25
  )
26
  text = ' '.join([summ['summary_text'] for summ in res])