Update app.py

app.py

@@ -27,16 +27,16 @@ def classify_potato_plant(img):
         model_output = "Early blight"
     elif class_labels[predicted_class] == "Potato__Late_blight":
         model_output = "Late blight"
-    elif class_labels[predicted_class] == "Potate__healthy":
+    elif class_labels[predicted_class] == "Potato__healthy":
         model_output = "Healthy"
 
-    return model_output, confidence
+    return class_labels[predicted_class], model_output, confidence
 
 # Create the Gradio interface
 interface = gr.Interface(
     fn=classify_potato_plant,
     inputs=gr.Image(type="pil"),
-    outputs=[gr.Label(label="Predicted output", num_top_classes=1), gr.Textbox(label="Confidence Score")],
+    outputs=[gr.Label(label="Predicted Classes", num_top_classes=1), gr.Textbox(label="Predicted Output"), gr.Textbox(label="Confidence Score")],
     title="Acres - PPDC",
     description="Acres PPDC, is our Potato Plant Disease Classification vision model, capable of accurately classifying potato plant disease, based on a single image."
 )