omm7 commited on
Commit
aba87c4
·
verified ·
1 Parent(s): f76e588

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -1,7 +1,8 @@
1
-
2
  from flask import Flask, request, jsonify
3
  import joblib
4
  import pandas as pd
 
 
5
 
6
  # Load the model and preprocessor
7
  model = joblib.load('boston_housing_model.pkl')
@@ -9,25 +10,31 @@ preprocessor = joblib.load('preprocessor.pkl')
9
 
10
  app = Flask(__name__)
11
 
 
 
 
 
12
  @app.route('/predict', methods=['POST'])
13
  def predict():
14
  try:
15
  data = request.json
 
 
16
  df = pd.DataFrame(data, index=[0])
17
 
18
- # Preprocess the data using the loaded preprocessor
19
- # Note: The preprocessor expects a DataFrame, so we transform it here.
20
  processed_data = preprocessor.transform(df)
21
 
22
  # Make prediction
23
  prediction = model.predict(processed_data)
 
24
 
25
  return jsonify({'prediction': prediction[0]})
26
  except Exception as e:
 
27
  return jsonify({'error': str(e)})
28
 
29
  if __name__ == '__main__':
30
- import os
31
  port = int(os.environ.get("PORT", 5000))
 
32
  app.run(host='0.0.0.0', port=port)
33
-
 
 
1
  from flask import Flask, request, jsonify
2
  import joblib
3
  import pandas as pd
4
+ import os
5
+ from datetime import datetime
6
 
7
  # Load the model and preprocessor
8
  model = joblib.load('boston_housing_model.pkl')
 
10
 
11
  app = Flask(__name__)
12
 
13
+ @app.before_first_request
14
+ def startup_message():
15
+ print(f"[{datetime.now()}] 🚀 Backend server is up and running on port {os.environ.get('PORT', 5000)}!")
16
+
17
  @app.route('/predict', methods=['POST'])
18
  def predict():
19
  try:
20
  data = request.json
21
+ print(f"[{datetime.now()}] 📩 Received prediction request: {data}")
22
+
23
  df = pd.DataFrame(data, index=[0])
24
 
25
+ # Preprocess the data
 
26
  processed_data = preprocessor.transform(df)
27
 
28
  # Make prediction
29
  prediction = model.predict(processed_data)
30
+ print(f"[{datetime.now()}] ✅ Prediction result: {prediction[0]}")
31
 
32
  return jsonify({'prediction': prediction[0]})
33
  except Exception as e:
34
+ print(f"[{datetime.now()}] ❌ Error: {str(e)}")
35
  return jsonify({'error': str(e)})
36
 
37
  if __name__ == '__main__':
 
38
  port = int(os.environ.get("PORT", 5000))
39
+ print(f"[{datetime.now()}] Starting Flask server...")
40
  app.run(host='0.0.0.0', port=port)