First push

Dockerfile

@@ -0,0 +1,10 @@
+FROM python:3.10
+
+WORKDIR /code
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

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

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+fastapi
+uvicorn