Update app.py

app.py

@@ -41,7 +41,6 @@ def main():
         gill_size = st.selectbox("gill size",options = unique_gill_size)
         gill_color = st.selectbox("gill color",options = unique_gill_color)
         stalk_shape = st.selectbox("stalk shape",options = unique_stalk_shape)
-        stalk_shape = st.selectbox("stalk shape",options = unique_stalk_shape)
         stalk_root = st.selectbox("stalk root",options = unique_stalk_root)
         stalk_surface_above_ring = st.selectbox("stalk surface above ring",options = unique_stalk_surface_above_ring)
         stalk_surface_below_ring = st.selectbox("stalk surface below ring",options = unique_stalk_surface_below_ring)
@@ -59,7 +58,28 @@ def main():
         
         clicked = st.form_submit_button("Predict income")
         if clicked:
-            result=model.predict(pd.DataFrame({'cap-shape':[cap_shape],'cap-surface':[cap_surface],'cap-color':[cap_color], 'bruises':[bruises], 'odor':[odor], 'gill-attachment':[gill_attachment], 'gill-spacing':[gill_spacing], 'gill-size':[gill_size], 'gill-color':[gill_color], 'stalk-shape':[stalk_shape], 'stalk-root':[stalk_root], 'stalk-surface-above-ring':[stalk_surface_above_ring], 'stalk-surface-below-ring':[stalk_surface_below_ring], 'stalk-color-above-ring':[stalk_color_above_ring], 'stalk-color-below-ring':[stalk_color_below_ring], 'veil-type':[veil_type], 'veil-color':[veil_color], 'ring-number':[ring_number], 'ring-type':[ring_type], 'spore-print-color':[spore_print_color], 'population':[population], 'habitat':[habitat]}))
+            result=model.predict(pd.DataFrame({'cap-shape':[cap_shape],
+                                               'cap-surface':[cap_surface],
+                                               'cap-color':[cap_color], 
+                                               'bruises':[bruises], 
+                                               'odor':[odor], 
+                                               'gill-attachment':[gill_attachment],
+                                               'gill-spacing':[gill_spacing], 
+                                               'gill-size':[gill_size], 
+                                               'gill-color':[gill_color], 
+                                               'stalk-shape':[stalk_shape], 
+                                               'stalk-root':[stalk_root], 
+                                               'stalk-surface-above-ring':[stalk_surface_above_ring], 
+                                               'stalk-surface-below-ring':[stalk_surface_below_ring], 
+                                               'stalk-color-above-ring':[stalk_color_above_ring], 
+                                               'stalk-color-below-ring':[stalk_color_below_ring], 
+                                               'veil-type':[veil_type], 
+                                               'veil-color':[veil_color], 
+                                               'ring-number':[ring_number], 
+                                               'ring-type':[ring_type], 
+                                               'spore-print-color':[spore_print_color], 
+                                               'population':[population], 
+                                               'habitat':[habitat]}))
             # Show prediction
             result = 'eatable' if result[0]==1 else 'poisonous'
             st.success("This Mushroom is "+result)