pratikdoshi commited on
Commit
64b7f01
·
verified ·
1 Parent(s): 26d7cd4

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # Load summarization model
5
+ @st.cache_resource
6
+ def load_summarizer():
7
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
8
+ return summarizer
9
+
10
+ # Summarize input text
11
+ def summarize_text(input_text, max_length=130, min_length=30):
12
+ summarizer = load_summarizer()
13
+
14
+ summary = summarizer(input_text, max_length=max_length, min_length=min_length, do_sample=False)
15
+
16
+ return summary[0]['summary_text']
17
+
18
+ # Streamlit app interface
19
+ def main():
20
+ st.title("Text Summarization App")
21
+
22
+ # Input box for text to summarize
23
+ input_text = st.text_area("Enter the text you want to summarize:", "Paste your article or text here...")
24
+
25
+ # Slider for max and min length of the summary
26
+ max_length = st.slider("Maximum length of summary:", min_value=50, max_value=300, value=130)
27
+ min_length = st.slider("Minimum length of summary:", min_value=20, max_value=100, value=30)
28
+
29
+ # Button to summarize
30
+ if st.button("Summarize"):
31
+ with st.spinner("Summarizing..."):
32
+ summary = summarize_text(input_text, max_length=max_length, min_length=min_length)
33
+ st.subheader("Summary")
34
+ st.write(summary)
35
+
36
+ # Run the app
37
+ if __name__ == '__main__':
38
+ main()