update app.py

app.py

@@ -38,7 +38,10 @@ def main():
     input_values = [sepal_length, sepal_width, petal_length, petal_width]
     prediction_result = predict_iris_species(trained_model, input_values)
 
-    st.write(f"Predicted Species: {prediction_result[0]}")
+    species_mapping = {0: 'Iris-setosa', 1: 'Iris-virginica', 2: 'Iris-versicolor'}
+    predicted_species = species_mapping.get(prediction_result[0], 'Unknown')
+
+    st.write(f"Predicted Species: {predicted_species}")
 
 if __name__ == "__main__":
     main()