Experimenting with few-shot classification.

app.py

@@ -51,8 +51,8 @@ model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
 classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer)
 
-few_shot_tokenizer = AutoTokenizer.from_pretrained('bge_auto_opt_O4', model_max_length=512)
-ort_model = ORTModelForFeatureExtraction.from_pretrained('bge_auto_opt_O4')
+few_shot_tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-small-en-v1.5', model_max_length=512)
+ort_model = ORTModelForFeatureExtraction.from_pretrained('BAAI/bge-small-en-v1.5')
 few_shot_model = SetFitModel.from_pretrained("moshew/bge-small-en-v1.5_setfit-sst2-english")
 onnx_few_shot_model = OnnxSetFitModel(ort_model, few_shot_tokenizer, few_shot_model.model_head)