Create app.py

app.py

@@ -0,0 +1,27 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
+
+MODEL_ID = "MoritzLaurer/deberta-v3-large-zeroshot-v2.0-c"
+app = FastAPI()
+zsc = pipeline("zero-shot-classification", model=MODEL_ID)
+
+class Req(BaseModel):
+    text: str
+    labels: list[str]
+    multi_label: bool = True
+    template: str = "This text is about {}."
+
+@app.get("/")
+def health():
+    return {"status": "ok"}
+
+@app.post("/predict")
+def predict(r: Req):
+    out = zsc(
+        r.text,
+        candidate_labels=r.labels,
+        multi_label=r.multi_label,
+        hypothesis_template=r.template,
+    )
+    return {"labels": out["labels"], "scores": out["scores"]}