Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def load_models():
|
|
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="
|
37 |
|
38 |
return models
|
39 |
|
|
|
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")
|
37 |
|
38 |
return models
|
39 |
|