Update app.py

app.py

@@ -4,6 +4,9 @@ import streamlit as st
 
 from PIL import Image
 image = Image.open('iris.png')
+image_setosa = Image.open('Irissetosa.jpg')
+image_virginica = Image.open('iris_virginica.jpg')
+image_versicolor = Image.open('versicolor.jpg')
 
 st.image(image, caption='Iris')
 model = joblib.load('model_XG.joblib')
@@ -34,11 +37,11 @@ def main():
             st.success(f"Your predicted class is {result}")
             
             if result[0] == 0:
-                st.image('iris.png')
+                st.image(image_setosa, caption='Iris Setosa')
             elif result[0] == 1:
-                st.image('iris.png')
+                st.image(image_versicolor, caption='Iris Versicolor')
             else:
-                st.image('iris.png')
+                st.image(image_virginica, caption='Iris Virginica')
                 
 if __name__ == '__main__':
     main()