Update app.py

app.py

@@ -6,18 +6,16 @@ import numpy as np
 
 app = Flask(__name__)
 
-saved_model_path = "superkart_model_v1_0.joblib" # Path inside Docker container
-loaded_model = joblib.load(saved_model_path)
+model = joblib.load("superkart_model.joblib")
 
-@app.route('/predict', methods=['POST'])
+@app.route("/")
+def health():
+    return "OK"
+
+@app.route("/v1/predict", methods=["POST"])
 def predict():
-    try:
-        json_ = request.json
-        query_df = pd.DataFrame(json_)
-        predictions = loaded_model.predict(query_df)
-        return jsonify({'predictions': predictions.tolist()})
-    except Exception as e:
-        return jsonify({'error': str(e)}), 400
+    data = request.get_json()
+    df = pd.DataFrame([data])
+    return jsonify({"Sales": model.predict(df).tolist()[0]})
 
-if __name__ == '__main__':
-    app.run(debug=True)
+app.run(host="0.0.0.0", port=7860)