Updated app.py

app.py

@@ -20,7 +20,10 @@ pipe = pipeline(
 )
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil"),
+    inputs=gr.Image(
+        type="pil",
+        height=400,
+    ),
     outputs=gr.AnnotatedImage(
         color_map={
             "Background (waterbody)": "#000000",
@@ -31,7 +34,8 @@ demo = gr.Interface(
             "Reefs and invertebrates": "#00FFFF",
             "Fish and vertebrates": "#FF00FF",
             "Sea-floor and rocks": "#FFFFFF",
-        }
+        },
+        height=400,
     ),
     examples="examples",
 )