Upload 3 files

Dockerfile.txt

@@ -0,0 +1,14 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+COPY api.py .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 8000
+
+CMD ["python", "api.py"]
+
+

app.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+app = FastAPI()
+
+classifier = pipeline("text-classification", model="mrm8488/t5-base-finetuned-emotion")
+
+class TextInput(BaseModel):
+    text: str
+
+@app.post("/predict")
+async def predict_emotion(input: TextInput):
+    result = classifier(input.text)
+    return {"emotion": result[0]['label'], "score": result[0]['score']}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn[standard]
+transformers
+torch