CR7CAD commited on
Commit
6200bd8
·
verified ·
1 Parent(s): 601e2aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def load_models():
30
  with st.spinner("Loading AI models... This may take a minute on first run."):
31
  models = {}
32
  # Load summarization model
33
- models['summarizer'] = pipeline("summarization", model="google/t5-small")
34
 
35
  # Load feature extraction model for similarity
36
  models['feature_extractor'] = pipeline("feature-extraction", model="bert-base-uncased")
 
30
  with st.spinner("Loading AI models... This may take a minute on first run."):
31
  models = {}
32
  # Load summarization model
33
+ models['summarizer'] = pipeline("summarization", model="sshleifer/distilbart-xsum-12-6")
34
 
35
  # Load feature extraction model for similarity
36
  models['feature_extractor'] = pipeline("feature-extraction", model="bert-base-uncased")