Update app.py

app.py

@@ -5,31 +5,30 @@ import streamlit as st
 
 model = joblib.load('model_XG.joblib')
 unique_values = joblib.load('unique_values_XG.joblib')
-    
 def main():
     st.title("Iris's Class")
-
+    
     with st.form("questionaire"):
-        sepal_length = st.slider("Sepal_length", min_value=0, max_value=6)
-        sepal_width = st.slider("Sepal_width", min_value=0, max_value=6)
-        petal_length = st.slider("Petal_length", min_value=0, max_value=6)
-        petal_width = st.slider("Petal_width", min_value=0, max_value=6)
+        sepal_length = st.slider("Sepal_length(cm)", 0.0, 10.0, 0.1)
+        sepal_width = st.slider("Sepal_width(cm)", 0.0, 10.0, 0.1)
+        petal_length = st.slider("Petal_length(cm)", 0.0, 10.0, 0.1)
+        petal_width = st.slider("Petal_width(cm)", 0.0, 10.0, 0.1)
 
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict class")
         if clicked:
-            result=model.predict(pd.DataFrame({"sepal_length": [sepal_length],
-                                               "sepal_width": [sepal_width],
-                                               "petal_length": [petal_length],
-                                               "petal_width": [petal_width]}))
+            result=model.predict(pd.DataFrame({"sepal.length": [sepal_length],
+                                               "sepal.width": [sepal_width],
+                                               "petal.length": [petal_length],
+                                               "petal.width": [petal_width]}))
             # Show prediction
-            if result[0] == 'Iris-setosa':
-                st.write("Iris-setosa")
-            elif result[0] == 'Iris-versicolor':
-                st.write("Iris-versicolor")
-            elif result[0] == 'Iris-virginica':
-                st.write("Iris-virginica")
+            if result[0] == 0:
+                result = "Iris-setosa"
+            elif result[0] == 1:
+                result = "Iris-versicolor"
+            else:
+                result = "Iris-virginica"
+            st.success(f"Your predicted class is {result}")
             
-            st.success("Your predicted class is"+result)
 if __name__ == "__main__":
     main()