Update app.py

app.py

@@ -111,6 +111,12 @@ def general_predictor(input_list):
   result = model.predict(new_input)
   return cat_value_dicts[final_colname][result[0]]
 
+def get_feat():
+  feats = [abs(x) for x in model.coef_[0]]
+  max_val = max(feats)
+  idx = feats.index(max_val)
+  return data.columns[idx]
+
 form = st.form('ml-inputs')
 
 # add data labels to replace those lost via star-args
@@ -136,9 +142,9 @@ if form.form_submit_button("Submit to get your recommendation!"):
   form.subheader(prediction)
   
 col1, col2, col3 = st.columns(3)
-col1.metric("Options", len(cat_value_dicts[final_colname]), "+2")
-col2.metric("Accuracy", round(metrics.accuracy_score(y_test, y_pred) * 100, 1), "-1%")
-col3.metric("Hi", 7)
+col1.metric("Number of Options", len(cat_value_dicts[final_colname]))
+col2.metric("Model Accuracy", str(round(metrics.accuracy_score(y_test, y_pred) * 100, 1)) + '%')
+col3.metric("Most Important Question", get_feat())
 
   
 with open('info.md') as f: