ojasrohatgi commited on
Commit
998932f
·
verified ·
1 Parent(s): 24419f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ from reportlab.lib.enums import TA_JUSTIFY
12
  import pyttsx3
13
 
14
  # Initialize the summarization pipeline
15
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
16
 
17
  # Set page layout to wide
18
  st.set_page_config(layout="wide")
@@ -45,7 +45,7 @@ def read_aloud(text):
45
 
46
  # Main application
47
  def main():
48
- st.title("Enhanced Article Extractor and Summarizer")
49
 
50
  url = st.text_input("Enter the URL of an article:", key="url")
51
  max_chunk = 300
 
12
  import pyttsx3
13
 
14
  # Initialize the summarization pipeline
15
+ summarizer = pipeline("summarization", model="facebook/bart-small") # facebook/bart-large-cnn
16
 
17
  # Set page layout to wide
18
  st.set_page_config(layout="wide")
 
45
 
46
  # Main application
47
  def main():
48
+ st.title("Abstractive Article Summarizer")
49
 
50
  url = st.text_input("Enter the URL of an article:", key="url")
51
  max_chunk = 300