Lord-Raven commited on
Commit
cfd4b0d
·
1 Parent(s): dc02763

Playing with other models.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -20,10 +20,11 @@ app.add_middleware(
20
  # "xenova/mobilebert-uncased-mnli" "typeform/mobilebert-uncased-mnli" Fast but small
21
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
22
  # "Xenova/bart-large-mnli" A bit slow
 
23
 
24
- model_name = "Xenova/distilbert-base-uncased-mnli"
25
  file_name = "onnx/model_quantized.onnx"
26
- tokenizer_name = "typeform/distilbert-base-uncased-mnli"
27
  model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
28
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
29
 
 
20
  # "xenova/mobilebert-uncased-mnli" "typeform/mobilebert-uncased-mnli" Fast but small
21
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
22
  # "Xenova/bart-large-mnli" A bit slow
23
+ # "Xenova/distilbert-base-uncased-mnli" "typeform/distilbert-base-uncased-mnli" Bad answers
24
 
25
+ model_name = "Xenova/deBERTa-v3-base-mnli"
26
  file_name = "onnx/model_quantized.onnx"
27
+ tokenizer_name = "MoritzLaurer/DeBERTa-v3-base-mnli"
28
  model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
29
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
30