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

Playing with other models.

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -17,11 +17,12 @@ app.add_middleware(
17
  allow_methods=["*"],
18
  allow_headers=["*"],
19
  )
 
20
 
21
- model_name = "xenova/deberta-v3-base-tasksource-nli" # "xenova/mobilebert-uncased-mnli"
22
  file_name = "onnx/model_quantized.onnx"
23
- tokenizer_name = "sileod/deberta-v3-base-tasksource-nli" # "typeform/mobilebert-uncased-mnli"
24
- model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
25
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
26
 
27
  # file = cached_download("https://huggingface.co/" + model_name + "")
 
17
  allow_methods=["*"],
18
  allow_headers=["*"],
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 + "")