Update models/summarization.py

models/summarization.py

@@ -0,0 +1,21 @@
+# models/summarization.py
+from typing import Any
+from models.utils import create_pipeline
+
+AVAILABLE_MODELS = {
+    # small/medium indonesian summarization model (placeholder)
+    "indobart": "cahya/indobart-large",
+}
+
+
+def predict(text: str, model_key: str) -> str:
+    model_id = AVAILABLE_MODELS.get(model_key)
+    if model_id is None:
+        raise ValueError(f"Unknown summarization model key: {model_key}")
+    p, err = create_pipeline("summarization", model_id)
+    if err:
+        raise RuntimeError(err)
+    out = p(text, max_length=200, min_length=30)
+    if isinstance(out, list) and out:
+        return out[0].get('summary_text', str(out[0]))
+    return str(out)