Update app.py

app.py

@@ -37,11 +37,11 @@ def main():
             st.success(f"Your predicted class is {result}")
             
             if result[0] == 0:
-                st.image(image_setosa, caption='Iris Setosa')
+                st.image(image_setosa, caption='Iris Setosa', width = 350)
             elif result[0] == 1:
-                st.image(image_versicolor, caption='Iris Versicolor')
+                st.image(image_versicolor, caption='Iris Versicolor', width = 350)
             else:
-                st.image(image_virginica, caption='Iris Virginica')
+                st.image(image_virginica, caption='Iris Virginica', width = 350)
                 
 if __name__ == '__main__':
     main()