Update app.py

app.py

@@ -6,12 +6,12 @@ from tensorflow.keras.preprocessing.image import img_to_array
 from PIL import Image
 import numpy as np
 
-# Load a stronger pretrained model (EfficientNetV2L)
+# Load the stronger pre-trained model (EfficientNetV2L)
 model = EfficientNetV2L(weights="imagenet")
 
 def predict_image(image):
     """
-    Process the uploaded image and return the top 5 predictions as a list.
+    Process the uploaded image and return the top 3 predictions as a dictionary.
     """
     try:
         # Preprocess the image
@@ -22,23 +22,22 @@ def predict_image(image):
 
         # Get predictions
         predictions = model.predict(image_array)
-        decoded_predictions = decode_predictions(predictions, top=5)[0]
+        decoded_predictions = decode_predictions(predictions, top=3)[0]
 
-        # Format predictions as a list of tuples (label, confidence)
-        results = [(label, float(confidence)) for _, label, confidence in decoded_predictions]
+        # Format predictions as a dictionary (label -> confidence)
+        results = {label: float(confidence) for _, label, confidence in decoded_predictions}
         return results
 
     except Exception as e:
-        return [("Error", str(e))]
+        return {"Error": str(e)}
 
-# Create the Gradio interface with gr.Dataframe for displaying predictions
+# Create the Gradio interface
 interface = gr.Interface(
     fn=predict_image,
     inputs=gr.Image(type="pil"),  # Accepts an image input
-    outputs=gr.Dataframe(headers=["Label", "Confidence"], type="array"),  # Shows top 5 predictions as a table
-    title="Image Classifier",
-    description="Upload an image, and the model will predict what's in the image with higher accuracy.",
-    examples=["dog.jpg", "cat.jpg", "building.jpg", "tree.jpg"],  # Example images for testing
+    outputs=gr.Label(num_top_classes=3),  # Shows top 3 predictions with confidence
+    title="EfficientNetV2L Image Classifier",
+    description="Upload an image, and the model will predict what's in the image with higher accuracy."
 )
 
 # Launch the Gradio app