Update app.py

app.py

@@ -1,88 +1,17 @@
-import joblib
-import pandas as pd
 import streamlit as st
+from transformers import pipeline
 
-model = joblib.load('model_tree.joblib')
-unique_values = joblib.load('unique_values.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"]
-
+classifier = pipeline("translation_en_to_fr",model="t5-base")
 def main():
-    st.title("Is this mushroom eatable")
-    with st.form("questionaire"):
-        cap_shape = st.selectbox("cap_shape",options = unique_cap_shape)
-        cap_surface = st.selectbox("cap_surface",options = unique_cap_surface)
-        cap_color = st.selectbox("cap_color",options = unique_cap_color)
-        bruises = st.selectbox("bruises",options = unique_bruises)
-        odor = st.selectbox("odor",options = unique_odor)     
-        gill_attachment = st.selectbox("gill_attachment",options = unique_gill_attachment)
-        gill_spacing = st.selectbox("gill_spacing",options = unique_gill_spacing)
-        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_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)
-        stalk_color_above_ring = st.selectbox("stalk_color_above_ring",options = unique_stalk_color_above_ring)
-        stalk_color_below_ring = st.selectbox("stalk_color_below_ring",options = unique_stalk_color_below_ring)
-        veil_type = st.selectbox("veil_type",options = unique_veil_type)
-        veil_color = st.selectbox("veil_color",options = unique_veil_color)
-        ring_number = st.selectbox("ring_number",options = unique_ring_number)
-        ring_type = st.selectbox("ring_type",options = unique_ring_type)
-        spore_print_color = st.selectbox("spore_print_color",options = unique_unique_spore_print_color)
-        population = st.selectbox("population",options = unique_population)
-        habitat = st.selectbox("habitat",options = unique_habitat)     
-        
-        
-        
-        clicked = st.form_submit_button("Predict income")
+    st.title("English to french")
+
+    with st.form("text_field"):
+        text = st.text_area('enter some text:')
+        # clicked==True only when the button is clicked
+        clicked = st.form_submit_button("Submit")
         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()
+          results = classifier([text])
+          st.json(results)
+
+if __name__ == "__main__":
+    main()