Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def load_models():
|
|
38 |
models['summarizer'] = pipeline("summarization", model="facebook/bart-large-cnn", max_length=130)
|
39 |
|
40 |
# Load smaller feature extraction model for speed
|
41 |
-
models['feature_extractor'] = pipeline("feature-extraction", model="
|
42 |
|
43 |
return models
|
44 |
|
|
|
38 |
models['summarizer'] = pipeline("summarization", model="facebook/bart-large-cnn", max_length=130)
|
39 |
|
40 |
# Load smaller feature extraction model for speed
|
41 |
+
models['feature_extractor'] = pipeline("feature-extraction", model="roberta-base")
|
42 |
|
43 |
return models
|
44 |
|