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]], cat_value_dicts[final_colname][result[1]]]
+  return cat_value_dicts[final_colname][result[0]]
 
 # add data labels to replace those lost via star-args
 inputls = []
@@ -129,10 +129,9 @@ for colname in data.columns:
   
 # generate gradio interface
 if st.button("Submit"):
-  results = general_predictor(inputls)
+  prediction = general_predictor(inputls)
   
-  st.text(f"Your most likley recommendation: {results[0]}")
-  st.text(f"Your second best recommendation: {results[1]}")
+  st.text(f"Our best recommendation for you: {prediction}")
   
 with open('info.md') as f:
   st.markdown(f.read())