Update app.py

app.py

@@ -0,0 +1,11 @@
+from fastapi import FastAPI
+from transformers import pipeline
+
+app = FastAPI()
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+@app.post("/predict")
+async def predict(data: dict):
+    labels = ["Mathematics", "Language Arts", "Social Studies", "Science"]
+    result = classifier(data["text"], labels)
+    return {"label": result["labels"][0]}