Update app.py

app.py

@@ -109,7 +109,7 @@ def general_predictor(input_list):
   # predict single datapoint
   new_input = [features]
   result = model.predict(new_input)
-  return cat_value_dicts[final_colname][result[0]]
+  return [cat_value_dicts[final_colname][result[0]], cat_value_dicts[final_colname][result[1]]]
 
 # add data labels to replace those lost via star-args
 inputls = []
@@ -129,9 +129,10 @@ for colname in data.columns:
   
 # generate gradio interface
 if col2.button("Submit"):
-  prediction = general_predictor(inputls)
+  results = general_predictor(inputls)
   
-  st.text(f"Your recommendation: {prediction}")
+  st.text(f"Your most likley recommendation: {results[0]}")
+  st.text(f"Your second best recommendation: {results[1]}")
   
 with open('info.md') as f:
   st.markdown(f.read())