Update model.py

model.py

@@ -184,7 +184,7 @@ detector.eval()
 
 def predict(code: str, threshold: float = THRESHOLD) -> dict:
     if len(code.strip()) < 200:
-        return {"prediction": "Too Short", "ai_probability": None}
+        return {"pred": "too short", "prob": None}
 
     with torch.no_grad():
         g      = gen_tokenizer(code, return_tensors="pt", padding="max_length", truncation=True, max_length=MAX_LEN).to(DEVICE)