Adiciona API XCOMET-XXL-app

app.py

@@ -0,0 +1,26 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from comet import download_model, load_from_checkpoint
+
+# Inicializa FastAPI
+app = FastAPI()
+
+# Carrega o modelo (uma vez só ao iniciar)
+model_path = download_model("Unbabel/XCOMET-XXL")
+model = load_from_checkpoint(model_path)
+
+# Define formato da requisição
+class TranslationPair(BaseModel):
+    src: str
+    mt: str
+    ref: str = None  # opcional
+
+@app.post("/score")
+def score_translation(pair: TranslationPair):
+    data = [{"src": pair.src, "mt": pair.mt, "ref": pair.ref}]
+    output = model.predict(data, batch_size=8, gpus=1)
+    return {
+        "segment_scores": output.scores,
+        "system_score": output.system_score,
+        "error_spans": getattr(output.metadata, "error_spans", None)
+    }