Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
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")
|