Create app/main.py

app/main.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from app.model import load_model, predict
+
+app = FastAPI()
+model, tokenizer, mlb = load_model()
+
+class PredictRequest(BaseModel):
+    text: str
+
+@app.post("/predict")
+async def predict_labels(request: PredictRequest):
+    text = request.text
+    pred_labels = predict(text, model, tokenizer, mlb)
+    return {"labels": pred_labels}
+
+@app.get("/")
+def root():
+    return {"message": "BERT Multi-label API is live"}