Update main.py

main.py

@@ -33,8 +33,11 @@ def home():
 def predict():
     try:
         # Get code from request
-        data = request.get_json()
-        if "code" not in data:
+        # data = request.get_json()
+        # if "code" not in data:
+        #     return jsonify({"error": "Missing 'code' parameter"}), 400
+        data = request.get_json(force=True, silent=True)
+        if not data or "code" not in data:
             return jsonify({"error": "Missing 'code' parameter"}), 400
             
         code = data["code"]