Update app.py

app.py

@@ -23,8 +23,6 @@ to_use_cols = ['as_of_year', 'county_name', 'applicant_sex_name', 'action_taken_
 
 data_reduced = data[to_use_cols]
 
-data_reduced.shape
-
 data_no_na = data_reduced.dropna()
 
 
@@ -37,14 +35,11 @@ mapped = {tuple(succeeded): 1, tuple(failed): 2, tuple(user_error): 3}
 def mapped(x):
   if x in succeeded:
     return 1
-  elif x in failed:
-    return 0
   else:
-    return 2
+    return 0
 
 data_no_na.action_taken_name = data_no_na.action_taken_name.apply(mapped)
 
-data_no_na.loan_type_name.value_counts()
 
 mapped_type = {
     'Conventional': 0,
@@ -126,7 +121,7 @@ with col1:
     submitted = st.form_submit_button("Submit")
     if submitted:
         col2.info('Predicting')
-        new_demand = np.array(mapping(county=county, sex=sex, property_type=property_type, loan=loan, income=income, loan_amount=loan_amount).values())
+        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))
         elif model_choice == 'RandomForest':