Experimenting with few-shot classification.

app.py

@@ -59,9 +59,10 @@ class OnnxSetFitModel:
 # "Xenova/bart-large-mnli" A bit slow
 # "Xenova/distilbert-base-uncased-mnli" "typeform/distilbert-base-uncased-mnli" Bad answers
 # "Xenova/deBERTa-v3-base-mnli" "MoritzLaurer/DeBERTa-v3-base-mnli" Still a bit slow and not great answers
-model_name = "xenova/nli-deberta-v3-small"
+# "xenova/nli-deberta-v3-small" "cross-encoder/nli-deberta-v3-small" Was using this for a good while and it was...okay
+model_name = "Xenova/deBERTa-v3-base-mnli"
 file_name = "onnx/model_quantized.onnx"
-tokenizer_name = "cross-encoder/nli-deberta-v3-small"
+tokenizer_name = "MoritzLaurer/DeBERTa-v3-base-mnli"
 model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
 classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer)