epalvarez commited on
Commit
3538277
·
verified ·
1 Parent(s): ebe1e38

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def predict_insurance_charge(age, bmi, children, sex, smoker, region):
81
  'sex': sex,
82
  'smoker': smoker,
83
  'region': region,
84
- 'prediction': prediction[0]
85
  }
86
  ))
87
  f.write("\n")
 
81
  'sex': sex,
82
  'smoker': smoker,
83
  'region': region,
84
+ 'prediction': prediction[0][0]
85
  }
86
  ))
87
  f.write("\n")