added images

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import pandas as pd
 from sklearn.model_selection import train_test_split
 from sklearn.neighbors import KNeighborsClassifier
+
 def train_iris_model():
     df = pd.read_csv("Iris.csv")
     df['Species'] = df['Species'].map({'Iris-setosa': 0, 'Iris-virginica': 1, 'Iris-versicolor': 2})
@@ -50,5 +51,13 @@ def main():
     st.subheader("Prediction:")
     st.success(f"Predicted Species: {predicted_species}")
 
+    # Display relevant images based on prediction
+    if predicted_species == 'Iris-setosa':
+        st.image('setosa_image.jpg', caption='Iris-setosa', use_column_width=True)
+    elif predicted_species == 'Iris-virginica':
+        st.image('virginica_image.jpg', caption='Iris-virginica', use_column_width=True)
+    elif predicted_species == 'Iris-versicolor':
+        st.image('versicolor_image.jpg', caption='Iris-versicolor', use_column_width=True)
+
 if __name__ == "__main__":
     main()