Create new file

app.py

@@ -0,0 +1,68 @@
+import joblib
+import pandas as pd
+import streamlit as st
+
+model = joblib.load('model_tree.joblib')
+unique_values = joblib.load('unique_value.joblib')
+    
+unique_cap-shape =  unique_values['cap-shape']
+unique_cap-surface =  unique_values["cap-surface"]
+unique_cap-color =  unique_values["cap-color"]
+unique_bruises =  unique_values["bruises"]
+unique_odor =  unique_values["odor"]
+unique_gill-attachment =  unique_values["gill-attachment"]
+unique_gill-spacing = unique_values["gill-spacing"]
+unique_gill-size =  unique_values["gill-size"]
+unique_gill-color =  unique_values['gill-color']
+unique_stalk-shape =  unique_values["stalk-shape"]
+unique_stalk-root =  unique_values["stalk-root"]
+unique_stalk-surface-above-ring =  unique_values["stalk-surface-above-ring"]
+unique_stalk-surface-below-ring =  unique_values["stalk-surface-below-ring"]
+unique_stalk-color-above-ring =  unique_values["stalk-color-above-ring"]
+unique_stalk-color-below-ring = unique_values["stalk-color-below-ring"]
+unique_veil_type =  unique_values["veil-type"]
+unique_veil-color =  unique_values['veil-color']
+unique_ring-number =  unique_values["ring-number"]
+unique_ring_type =  unique_values["ring-type"]
+unique_spore_print-color =  unique_values["spore-print-color"]
+unique_population =  unique_values["population"]
+unique_habitat =  unique_values["habitat"]
+
+def main():
+    st.title("Is this mushroom eatable")
+    with st.form("questionaire"):
+        cap_shape = st.selectbox("cap shape",option = unique_cap-shape)
+        cap_surface = st.selectbox("cap surface",option = unique_cap-surface)
+        cap-color = st.selectbox("cap color",option = unique_cap-color)
+        bruises = st.selectbox("bruises",option = unique_bruises)
+        odor = st.selectbox("odor",option = unique_odor)     
+        gill-attachment = st.selectbox("gill attachment",option = unique_gill-attachment)
+        gill-spacing = st.selectbox("gill spacing",option = unique_gill-spacing)
+        gill-size = st.selectbox("gill size",option = unique_gill-size)
+        gill-color = st.selectbox("gill color",option = unique_gill-color)
+        stalk-shape = st.selectbox("stalk shape",option = unique_stalk-shape)
+        stalk-shape = st.selectbox("stalk shape",option = unique_stalk-shape)
+        stalk-root = st.selectbox("stalk root",option = unique_stalk-root)
+        stalk-surface-above-ring = st.selectbox("stalk surface above ring",option = unique_stalk-surface-above-ring)
+        stalk-surface-below-ring = st.selectbox("stalk surface below ring",option = unique_stalk-surface-below-ring)
+        stalk-color-above-ring = st.selectbox("stalk color above ring",option = unique_stalk-color-above-ring)
+        stalk-color-below-ring = st.selectbox("stalk color below ring",option = unique_stalk-color-below-ring)
+        veil_type = st.selectbox("veil type",option = unique_veil_type)
+        veil-color = st.selectbox("veil color",option = unique_veil-color)
+        ring-number = st.selectbox("ring number",option = unique_ring-number)
+        ring_type = st.selectbox("ring type",option = unique_ring_type)
+        spore_print-color = st.selectbox("spore print color",option = unique_unique_spore_print-color)
+        population = st.selectbox("population",option = unique_population)
+        habitat = st.selectbox("habitat",option= unique_habitat)     
+        
+        
+        
+        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]}))
+            # Show prediction
+            result = 'eatable' if result[0]==1 else 'poisonous'
+            st.success("This Mushroom is "+result)
+# Run main()
+if __name__=="__main__":
+  main()