Update main.py

main.py

@@ -54,8 +54,8 @@ def predict():
         score = torch.sigmoid(outputs.logits).item()
 
         return jsonify({
-            "readability_score": round(score, 4),
-            "processed_code": code[:500] + "..." if len(code) > 500 else code
+            "score": round(score, 4),
+            "given_code": code[:500] + "..." if len(code) > 500 else code
         })
 
     except Exception as e: