s

app.py

@@ -12,4 +12,9 @@ def greet_json():
         "This is a course about the Transformers library",
         candidate_labels=["education", "politics", "business"],
     )
-    return {"Hello": str(j)}
+    return {"Hello": str(j)}
+
+@app.post('/zero-shot-classification')
+def classify(source: str, labels: list[str]) -> dict[str, str]:
+    c = classifier(source, candidate_labels = labels)
+    return dict(zip(c['labels'], c['scores']))