Create main.py

main.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from transformers import pipeline
+
+app = FastAPI()
+
+# Load model once when the app starts
+classifier = pipeline("zero-shot-classification", model="syedkhalid076/DeBERTa-Zero-Shot-Classification")
+
+# Request body model
+class ClassificationRequest(BaseModel):
+    text: str
+
+@app.post("/classify")
+async def classify(request: ClassificationRequest):
+    candidate_labels = ["contains electronic components", "doesn't contain electronic components"]
+    result = classifier(request.text, candidate_labels, multi_label=False)
+    return {"labels": result["labels"], "scores": result["scores"]}