Lord-Raven commited on
Commit
cc8b299
·
1 Parent(s): 6c40a85

Playing with other models.

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,10 +19,10 @@ app.add_middleware(
19
  )
20
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
21
 
22
- model_name = "MoritzLaurer/deberta-v3-large-zeroshot-v2.0" # "xenova/mobilebert-uncased-mnli"
23
  file_name = "onnx/model_quantized.onnx"
24
- tokenizer_name = model_name # "typeform/mobilebert-uncased-mnli"
25
- model = ORTModelForSequenceClassification.from_pretrained(model_name)
26
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
27
 
28
  # file = cached_download("https://huggingface.co/" + model_name + "")
 
19
  )
20
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
21
 
22
+ model_name = "Xenova/bart-large-mnli" # "xenova/mobilebert-uncased-mnli"
23
  file_name = "onnx/model_quantized.onnx"
24
+ tokenizer_name = "facebook/bart-large-mnli" # "typeform/mobilebert-uncased-mnli"
25
+ model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
26
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
27
 
28
  # file = cached_download("https://huggingface.co/" + model_name + "")