3

app.py

@@ -0,0 +1,20 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
+
+app = FastAPI()
+
+@app.get("/")
+def root():
+    return {"status": "ok"}
+
+class Query(BaseModel):
+    text: str
+
+@app.post("/predict")
+def predict(query: Query):
+    result = classifier(query.text)[0]
+    return {"label": result["label"], "score": result["score"]}
+