Update app.py

app.py

@@ -54,27 +54,19 @@ with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
     gr.Markdown(description)
     
-    with gr.Row():
-        model_dropdown = gr.Dropdown(choices=model_names, label="Select Model", value=model_names[0])
-        image_input = gr.Image(type="pil")
     
-    # Move the Examples section here, before the output
+    model_dropdown = gr.Dropdown(choices=model_names, label="Select Model", value=model_names[0])
+    image_input = gr.Image(type="pil")
+    
     gr.Examples(
         examples=example_images,
         inputs=image_input,
         label="Click on an example image to test",
     )
     
-    # Output section
     output = gr.Label(label="Classification Result")
-
-    # Event handlers
-    def classify(img, model):
-        if img is not None:
-            return predict(img, model)
-        return None
-
-    image_input.change(fn=classify, inputs=[image_input, model_dropdown], outputs=output)
-    model_dropdown.change(fn=classify, inputs=[image_input, model_dropdown], outputs=output)
+    
+    image_input.change(fn=predict, inputs=[image_input, model_dropdown], outputs=output)
+    model_dropdown.change(fn=predict, inputs=[image_input, model_dropdown], outputs=output)
 
 demo.launch()