Update app.py

app.py

@@ -21,12 +21,17 @@ except:
 @app.route("/predict", methods=["POST"])
 def predict():
     data = request.json
-    features = [data["sepal_length"], data["sepal_width"],
-                data["petal_length"], data["petal_width"]]
-    prediction = model.predict([features])[0]
+    features = [
+        data["sepal_length"],
+        data["sepal_width"],
+        data["petal_length"],
+        data["petal_width"]
+    ]
 
+    prediction = model.predict([features])[0]  # Get scalar (0,1,2)
     target_names = load_iris().target_names
-    return jsonify({"prediction": target_names[prediction]})
+    return jsonify({"prediction": target_names[int(prediction)]})
+
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)