fix: add app logic and Docker config

Dockerfile

@@ -1,11 +1,10 @@
-FROM python:3.11-slim
+FROM python:3.10
 
 WORKDIR /app
 
-COPY . .
-
+COPY requirements.txt .
 RUN pip install -r requirements.txt
 
-EXPOSE 7860
+COPY . .
 
 CMD ["python", "app.py"]

app.py

@@ -1,20 +1,22 @@
 from flask import Flask, request, jsonify
 import pickle
-import numpy as np
-import joblib
 
 app = Flask(__name__)
 
 # Load model
-model = joblib.load("models/model.pkl")
+with open("models/model.pkl", "rb") as f:
+    model = pickle.load(f)
 
 @app.route("/")
-def home():
-    return "API Iris Flower Ready!"
+def index():
+    return "Iris Classifier API is running!"
 
 @app.route("/predict", methods=["POST"])
 def predict():
-    data = request.json
-    features = np.array(data["features"]).reshape(1, -1)
-    prediction = model.predict(features)[0]
-    return jsonify({"prediction": prediction})
+    data = request.get_json()
+    features = [data["sepal_length"], data["sepal_width"], data["petal_length"], data["petal_width"]]
+    prediction = model.predict([features])
+    return jsonify({"prediction": prediction[0]})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

requirements.txt

@@ -1,6 +1,6 @@
 flask
 numpy
-scikit-learn
+scikit-learn==1.6.1
 joblib
-flask-cors
+flask
 gunicorn