Update app.py

app.py

@@ -122,16 +122,17 @@ def main():
     st.write(user_input)
     
     prediction = model.predict(user_input)
-    prediction_proba = model.predict(user_input, raw_score=True)
-
+  
     st.subheader('Prediction:')
-    if prediction[0] == 1:
-        st.write('The model predicts: Income > $50K')
-    else:
+    if prediction_proba[0] < 0.5:
         st.write('The model predicts: Income <= $50K')
+    else:
+        st.write('The model predicts: Income > $50K')
 
     st.subheader('Prediction Probability:')
     st.write(f'Probability of making over $50K: {prediction_proba[0]:.2f}')
 
 if __name__ == '__main__':
     main()
+if __name__ == '__main__':
+    main()