Update app.py

app.py

@@ -13,10 +13,11 @@ from sklearn import datasets
 iris = datasets.load_iris()
 
 # Define the prediction function
-def predict_iris(sepal_length, sepal_width, petal_length, petal_width):
+def predict_iris(sepal_length: "Sepal Length", sepal_width: "Sepal Width", petal_length: "Petal Length", petal_width: "Petal Width"):
     prediction = clf.predict([[sepal_length, sepal_width, petal_length, petal_width]])
     return iris.target_names[int(prediction[0])]
 
+
 # Create and launch the Gradio interface
 interface = gr.Interface(
     fn=predict_iris,