Create api.py

api.py

@@ -0,0 +1,18 @@
+# api.py
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import List, Dict, Any
+from model_utils import predict_proba
+
+app = FastAPI(title="StackOverflow Tagger API")
+
+
+class Query(BaseModel):
+    text: str
+    top_k: int = 10
+
+
+@app.post("/predict")
+def predict(q: Query) -> Dict[str, Any]:
+    tags = predict_proba(q.text, top_k=q.top_k)
+    return {"tags": tags}