Update app.py

app.py

@@ -28,7 +28,13 @@ def main():
                                                "petal_length": [petal_length],
                                                "petal_width": [petal_width]}))
             # Show prediction
-            result = "Iris-setosa" if result[0] == 1 , "Iris-versicolor" if result[0] == 2 else "Iris-virginica"
+            if result[0] == 1:
+                st.write("Iris-setosa")
+            elif result[0] == 2:
+                st.write("Iris-versicolor")
+            else:
+                st.write("Iris-virginica")
+            
             st.success("Your predicted class is"+result)
 if __name__ == "__main__":
     main()