Update app.py

app.py

@@ -95,13 +95,13 @@ with gr.Blocks() as demo:
         url_input = gr.Textbox(label="Image URL", placeholder="https://api.digitale-sammlungen.de/iiif/image/v2/bsb10483966_00008/full/500,/0/default.jpg")
     
     with gr.Tab("From Image"):
-        image_input = gr.Gallery(type="numpy", label="Input Image")
+        image_input = gr.Image(type="numpy", label="Input Image")
 
     min_score = gr.Slider(minimum=0.0, maximum=1.0, value=0.5, label="Minimum score")
 
     model_name = gr.Radio(choices=[model["name"] for model in models], value=models[0]["name"], label="Select Detectron2 model")
 
-    output_image = gr.Gallery(type="pil", label="Output")
+    output_image = gr.Image(type="pil", label="Output")
 
     inference_button = gr.Button("Submit")