app

app.py

@@ -1,11 +1,16 @@
 from flask import Flask, request, jsonify
 import joblib
+import json
 
 app = Flask(__name__)
 
 # Load model
 model = joblib.load("models/model.pkl")
 
+# Load label mapping
+with open("models/labels.json", "r") as f:
+    label_map = json.load(f)
+
 @app.route("/")
 def index():
     return "Iris Classifier API is running!"
@@ -21,6 +26,11 @@ def predict():
             float(data["petal_width"]),
         ]
         prediction = model.predict([features])
-        return jsonify({"prediction": int(prediction[0])})  # <-- Fix disini
+        pred_index = int(prediction[0])
+        pred_label = label_map.get(str(pred_index), "unknown")
+        return jsonify({
+            "prediction": pred_index,
+            "label": pred_label
+        })
     except Exception as e:
         return jsonify({"error": str(e)}), 400

models/labels.json

@@ -0,0 +1,5 @@
+{
+  "0": "Iris-setosa",
+  "1": "Iris-versicolor",
+  "2": "Iris-virginica"
+}