Update app.py

app.py

@@ -123,10 +123,15 @@ with col1:
         col2.info('Predicting')
         new_demand = np.array([list(mapping(county=county, sex=sex, property_type=property_type, loan=loan, income=income, loan_amount=loan_amount).values())])
         if model_choice == 'Tree':
-          col2.markdown(tree.predict(new_demand))
+          result = tree.predict(new_demand)
         elif model_choice == 'RandomForest':
-          col2.markdown(random.predict(new_demand))
+          result = random.predict(new_demand)
         else:
-          col2.text(gbm.predict(new_demand))
+          result = gbm.predict(new_demand)
+
+        if result[0]:
+            st.success('Accepted')
+        else:
+            st.failed("Rejected")