Update app.py

app.py

@@ -1,30 +1,15 @@
 import gradio as gr
 import pickle
 
-# Load model
 model = pickle.load(open("model.pkl", "rb"))
 
-# Flower labels
 flowers = ["Setosa", "Versicolor", "Virginica"]
 
-# Image URLs (you can also store locally)
-flower_images = {
-    "Setosa": "",
-    "Versicolor": "https://upload.wikimedia.org/wikipedia/commons/4/41/Iris_versicolor_3.jpg",
-    "Virginica": "https://upload.wikimedia.org/wikipedia/commons/9/9f/Iris_virginica.jpg"
-}
-
-
 def predict(sl, sw, pl, pw):
     data = [[sl, sw, pl, pw]]
     pred = model.predict(data)[0]
-    
-    flower_name = flowers[pred]
-    image = flower_images[flower_name]
-    
-    return flower_name, image
+    return flowers[pred]
 
-# Gradio Interface
 interface = gr.Interface(
     fn=predict,
     inputs=[
@@ -33,11 +18,8 @@ interface = gr.Interface(
         gr.Number(label="Petal Length"),
         gr.Number(label="Petal Width")
     ],
-    outputs=[
-        gr.Text(label="Prediction"),
-        gr.Image(label="Flower Image")
-    ],
-    title="🌸 Flower Prediction with Image"
+    outputs="text",
+    title="🌸 Flower Prediction App"
 )
 
 interface.launch()