Update models/qa.py

models/qa.py

@@ -0,0 +1,19 @@
+# models/qa.py
+
+from typing import Dict, Any
+from models.utils import create_pipeline
+
+AVAILABLE_MODELS = {
+    "indobert-qa": "indobenchmark/indobert-base-p1",
+}
+
+
+def predict(context: str, question: str, model_key: str) -> Dict[str, Any]:
+    model_id = AVAILABLE_MODELS.get(model_key)
+    if model_id is None:
+        raise ValueError(f"Unknown QA model key: {model_key}")
+    p, err = create_pipeline("question-answering", model_id)
+    if err:
+        raise RuntimeError(err)
+    out = p(question=question, context=context)
+    return out