Update app.py

app.py

@@ -29,15 +29,15 @@ def classify_image(img):
     # Format the output
     return {label: float(probability) for (_, label, probability) in decoded_predictions}
 
-# Gradio interface
-iface = gr.Interface(
-    fn=classify_image,  # Function to call for predictions
-    inputs=gr.Image(type="pil"),  # Input is an image
-    outputs=gr.Label(num_top_classes=3),  # Output is a label with top 3 predictions
-    title="Contextual Image Classification",
-    description="Upload an image, and the model will classify it based on the context.",
-    #examples=[chameleon],
-)
+with gr.Blocks(theme="Hev832/Applio") as iface:
+    gr.Interface(
+        fn=classify_image,  # Function to call for predictions
+        inputs=gr.Image(type="pil"),  # Input is an image        
+        outputs=gr.Label(num_top_classes=3),  # Output is a label with top 3 predictions
+        title="Contextual Image Classification",
+        description="Upload an image, and the model will classify it based on the context.",
+        #examples=[chameleon],
+    )
 
 # Launch the interface
 iface.launch()