Update model.py

model.py

@@ -194,6 +194,6 @@ def predict(code: str, threshold: float = THRESHOLD) -> dict:
         prob   = torch.sigmoid(logits).item()
 
     return {
-        "prediction": "AI Generated" if prob >= threshold else "Human Written",
-        "ai_probability": round(prob * 100, 2)
+        "pred": "ai-generated" if prob >= threshold else "human",
+        "prob": (prob * 100, 2)
     }