sathvikk commited on
Commit
8b08384
·
verified ·
1 Parent(s): 0a190ad

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -1
src/streamlit_app.py CHANGED
@@ -11,7 +11,8 @@ st.set_page_config(page_title="PrepPal", page_icon="📘", layout="wide")
11
  # Load summarizer model
12
  @st.cache_resource
13
  def load_summarizer():
14
- return pipeline("summarization", model="t5-small")
 
15
 
16
  # Extract text from uploaded PDF
17
  def extract_text_from_pdf(uploaded_file):
 
11
  # Load summarizer model
12
  @st.cache_resource
13
  def load_summarizer():
14
+ return pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
15
+
16
 
17
  # Extract text from uploaded PDF
18
  def extract_text_from_pdf(uploaded_file):