Xylor commited on
Commit
5843201
·
verified ·
1 Parent(s): 0ea8185

Revert to orig bert-base-uncased

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # 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"
 
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"