Spaces:
Running
Running
Revert to orig bert-base-uncased
Browse files
app.py
CHANGED
@@ -16,9 +16,9 @@ categories = [
|
|
16 |
logging.debug("Categories to classify: " + repr(categories))
|
17 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
18 |
classifier = pipeline("zero-shot-classification",
|
19 |
-
|
20 |
# model="facebook/bart-large-mnli")
|
21 |
-
model="Xylor/bert-case-uncased-single-sampled",
|
22 |
token=HF_TOKEN)
|
23 |
|
24 |
# sequence_to_classify = "one day I will see the world"
|
|
|
16 |
logging.debug("Categories to classify: " + repr(categories))
|
17 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
18 |
classifier = pipeline("zero-shot-classification",
|
19 |
+
model="google-bert/bert-base-uncased",
|
20 |
# model="facebook/bart-large-mnli")
|
21 |
+
# model="Xylor/bert-case-uncased-single-sampled",
|
22 |
token=HF_TOKEN)
|
23 |
|
24 |
# sequence_to_classify = "one day I will see the world"
|