Update app.py

app.py

@@ -28,10 +28,17 @@ def predict():
         data["petal_width"]
     ]
 
-    prediction = model.predict([features])[0]  # Get scalar (0,1,2)
+    prediction = model.predict([features])[0]  # could be int or string
+
+    # If prediction is already a string (class label)
+    if isinstance(prediction, str):
+        return jsonify({"prediction": prediction})
+
+    # If prediction is numeric (0, 1, 2)
     target_names = load_iris().target_names
     return jsonify({"prediction": target_names[int(prediction)]})
 
 
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)