Messing with configuration.

README.md

@@ -4,7 +4,7 @@ emoji: ⚡
 colorFrom: blue
 colorTo: yellow
 sdk: gradio
-sdk_version: 4.41.0
+sdk_version: 5.9.1
 app_file: app.py
 pinned: false
 license: openrail

app.py

@@ -7,10 +7,6 @@ from transformers import pipeline
 from optimum.onnxruntime import ORTModelForSequenceClassification
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from setfit import SetFitModel, SetFitTrainer, Trainer, TrainingArguments
-from setfit.exporters.utils import mean_pooling
-from setfit import get_templated_dataset
-from datasets import load_dataset, Dataset
 
 # CORS Config
 app = FastAPI()

requirements.txt

@@ -3,6 +3,5 @@ huggingface_hub==0.23.5
 json5==0.9.25
 numpy<2.0
 optimum[exporters,onnxruntime-gpu]==1.21.4
-setfit==1.0.3
-transformers==4.40.2
+transformers
 sentence-transformers==3.0.1