keerthas commited on
Commit
5a4ba5e
·
verified ·
1 Parent(s): f3941fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,7 +2,7 @@ from flask import Flask, request, jsonify
2
  import joblib
3
  import pandas as pd
4
 
5
- # Load model
6
  model = joblib.load("best_model.pkl")
7
 
8
  app = Flask(__name__)
@@ -15,12 +15,11 @@ def home():
15
  def predict():
16
  try:
17
  data = request.get_json()
18
- df = pd.DataFrame([data])
19
  prediction = model.predict(df)[0]
20
  return jsonify({"prediction": float(prediction)})
21
  except Exception as e:
22
  return jsonify({"error": str(e)})
23
 
24
  if __name__ == "__main__":
25
- app.run(host="0.0.0.0", port=7860)
26
-
 
2
  import joblib
3
  import pandas as pd
4
 
5
+ # Load trained model
6
  model = joblib.load("best_model.pkl")
7
 
8
  app = Flask(__name__)
 
15
  def predict():
16
  try:
17
  data = request.get_json()
18
+ df = pd.DataFrame([data]) # Convert JSON to DataFrame
19
  prediction = model.predict(df)[0]
20
  return jsonify({"prediction": float(prediction)})
21
  except Exception as e:
22
  return jsonify({"error": str(e)})
23
 
24
  if __name__ == "__main__":
25
+ app.run(host="0.0.0.0", port=7860) # Hugging Face expects 7860