Update app.py

app.py

@@ -27,11 +27,10 @@ def main():
             # Show prediction
             if result[0] == 0:
                 result = "Iris-setosa"
+                st.image(image, caption='Iris')
             elif result[0] == 1:
                 result = "Iris-versicolor"
             else:
                 result = "Iris-virginica"
             st.success(f"Your predicted class is {result}")
             
-if __name__ == "__main__":
-    main()