Modify request type + take params from JSON body.

main.py

@@ -29,13 +29,14 @@ 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
+        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(
@@ -53,10 +54,7 @@ def predict():
         # Apply sigmoid and format score
         score = torch.sigmoid(outputs.logits).item()
 
-        return jsonify({
-            "score": round(score, 4),
-            "given_code": code[:500] + "..." if len(code) > 500 else code
-        })
+        return jsonify(round(score, 4)) 
 
     except Exception as e:
         return jsonify({"error": str(e)}), 500