MrGanesh commited on
Commit
f924131
Β·
1 Parent(s): 35f58f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -5,20 +5,20 @@ from transformers import pipeline
5
  import streamlit as st
6
 
7
  def app():
8
- st.title("Text Summarization πŸ€“")
9
 
10
- st.markdown("This is a Web application that Summarizes Text 😎")
11
  upload_file = st.file_uploader('Upload a file containing Text data')
12
  button = st.button("Summarize")
13
 
14
  st.cache(allow_output_mutation=True)
15
- def facebook_bart_model():
16
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
17
  return summarizer
18
- summarizer= facebook_bart_model()
19
 
20
  def text_summarizer(text):
21
- a = summarizer(text, max_length=150, min_length=30, do_sample=False)
22
  return a[0]['summary_text']
23
 
24
 
 
5
  import streamlit as st
6
 
7
  def app():
8
+ st.title("Patent Document Summarization πŸ€“")
9
 
10
+ st.markdown("This is a Web application that Summarizes Patent Text 😎")
11
  upload_file = st.file_uploader('Upload a file containing Text data')
12
  button = st.button("Summarize")
13
 
14
  st.cache(allow_output_mutation=True)
15
+ def model():
16
+ summarizer = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
17
  return summarizer
18
+ summarizer= model()
19
 
20
  def text_summarizer(text):
21
+ a = summarizer(text, max_length=450, min_length=150, do_sample=False)
22
  return a[0]['summary_text']
23
 
24