Modify request type + take params from JSON body.

main.py

@@ -29,13 +29,15 @@ except Exception as e:
 def home():
     return request.url
 
-@app.route("/predict")
+@app.route("/predict", methods=["POST"])
 def predict():
     try:
-        # Get code from URL parameter
-        code = request.args.get("code")
-        if not code:
-            return jsonify({"error": "Missing 'code' URL parameter"}), 400
+        # Get code from request body (JSON)
+        data = request.get_json()
+        if not data or "code" not in data:
+            return jsonify({"error": "Missing 'code' in request body"}), 400
+
+        code = data["code"]
 
         # Tokenize input
         inputs = tokenizer(
@@ -54,8 +56,7 @@ def predict():
         score = torch.sigmoid(outputs.logits).item()
 
         return jsonify({
-            "score": round(score, 4),
-            "given_code": code[:500] + "..." if len(code) > 500 else code
+            "score": round(score, 4)
         })
 
     except Exception as e: