wiaandev commited on
Commit
c95da78
1 Parent(s): 8c474e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -21,5 +21,4 @@ for feature in model_info["features"]:
21
  if st.button("Predict Salary"):
22
  input_features = [feature_values[feature] for feature in model_info["features"]]
23
  prediction = model.predict([input_features])[0]
24
- rounded_prediction = math.ceil(prediction)
25
- st.write(f"Predicted annual salary: ${rounded_prediction[0]}")
 
21
  if st.button("Predict Salary"):
22
  input_features = [feature_values[feature] for feature in model_info["features"]]
23
  prediction = model.predict([input_features])[0]
24
+ st.write(f"Predicted annual salary: ${prediction[0]}")