Experimenting with few-shot classification.

app.py

@@ -4,6 +4,7 @@ import torch
 from transformers import AutoTokenizer
 from transformers import pipeline
 from optimum.onnxruntime import ORTModelForSequenceClassification
+from optimum.onnxruntime import ORTModelForFeatureExtraction
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from setfit import SetFitModel