Lord-Raven commited on
Commit
f4c9eb8
·
1 Parent(s): 3b93450

Playing with models.

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,11 +9,11 @@ from optimum.onnxruntime import ORTModelForSequenceClassification
9
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
10
  # "Xenova/bart-large-mnli" A bit slow
11
  # "Xenova/distilbert-base-uncased-mnli" "typeform/distilbert-base-uncased-mnli" Bad answers
12
-
13
- model_name = "Xenova/deBERTa-v3-base-mnli"
14
  file_name = "onnx/model_quantized.onnx"
15
- tokenizer_name = "MoritzLaurer/DeBERTa-v3-base-mnli"
16
- model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
17
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
18
 
19
  # file = cached_download("https://huggingface.co/" + model_name + "")
 
9
  # "xenova/deberta-v3-base-tasksource-nli" Not impressed
10
  # "Xenova/bart-large-mnli" A bit slow
11
  # "Xenova/distilbert-base-uncased-mnli" "typeform/distilbert-base-uncased-mnli" Bad answers
12
+ # "Xenova/deBERTa-v3-base-mnli" "MoritzLaurer/DeBERTa-v3-base-mnli" Still a bit slow and not great answers
13
+ model_name = "MoritzLaurer/deberta-v3-large-zeroshot-v2.0"
14
  file_name = "onnx/model_quantized.onnx"
15
+ tokenizer_name = model_name
16
+ model = ORTModelForSequenceClassification.from_pretrained(model_name) #, file_name=file_name)
17
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
18
 
19
  # file = cached_download("https://huggingface.co/" + model_name + "")