latest

.gitignore

@@ -1,4 +1,6 @@
 .env
 .venv/
 env/
-venv/
+venv/
+.__pycache__
+__pycache__/

app.py

@@ -4,19 +4,19 @@ from setfit import SetFitModel
 
 app = FastAPI()
 
+
 class TextInput(BaseModel):
     text: str
 
-class LabelScore(BaseModel):
-    label: str
-    score: float
 
-@app.post("/predict", response_model=LabelScore)
+@app.post("/predict")
 async def predict(input: TextInput):
+    text_input = input.text
     model = SetFitModel.from_pretrained("assets")
-    preds = model.predict(input)
-    return preds
+    preds = model.predict(text_input)
+    return {"label": preds}
+
 
 @app.get("/")
 def read_root():
-    return {"message": "Welcome to the FastAPI prediction service"}
+    return {"message": "Welcome"}