Update app.py

app.py

@@ -48,8 +48,6 @@ def predict_disease(image):
     """
     Preprocesses the image, performs prediction, and returns the predicted class with confidence score.
     """
-    # Resize and preprocess image
-    # Resize and preprocess the image
     img = tf.image.resize(image, [256, 256])
     img = tf.expand_dims(img, axis=0)  # Add batch dimension
 
@@ -60,11 +58,12 @@ def predict_disease(image):
     # Create a dictionary of class probabilities
     class_probs = {label: float(prob) for label, prob in zip(labels, prediction)}
 
+    return class_probs
 # ... (Rest of your code) ...
 
 iface = gr.Interface(
     fn=predict_disease,
-    inputs=gr.Image(type="numpy", tool="editor"),  # Allow image editing
+    inputs=gr.Image(),  # Allow image editing
     outputs=gr.Label(num_top_classes=3),          # Display all three probabilities
     title="Potato Disease Classifier 🍟🥔",
     description=(