Spaces:
Paused
Paused
Update nlp_module.py
Browse files- nlp_module.py +1 -1
nlp_module.py
CHANGED
|
@@ -12,7 +12,7 @@ def get_classifier():
|
|
| 12 |
if _classifier is None:
|
| 13 |
_classifier = pipeline(
|
| 14 |
"text-classification",
|
| 15 |
-
model="
|
| 16 |
top_k=None
|
| 17 |
)
|
| 18 |
return _classifier
|
|
|
|
| 12 |
if _classifier is None:
|
| 13 |
_classifier = pipeline(
|
| 14 |
"text-classification",
|
| 15 |
+
model="distilbert-base-uncased-finetuned-sst-2-english",
|
| 16 |
top_k=None
|
| 17 |
)
|
| 18 |
return _classifier
|