killswitch009 commited on
Commit
7d12025
·
verified ·
1 Parent(s): 5e80af1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def home():
18
  def predict():
19
 
20
  data = request.json["data"]
21
- prediction = model.predict(np.array(data).reshape(1, -1))
22
 
23
  return jsonify({
24
  "prediction": int(prediction[0])
 
18
  def predict():
19
 
20
  data = request.json["data"]
21
+ prediction = model.predict(np.zeros((1, model.n_features_in_)))
22
 
23
  return jsonify({
24
  "prediction": int(prediction[0])