Limit output image height

app.py

@@ -16,12 +16,12 @@ def predict(img, model_type):
 
 # Interface
 inputs = [
-    "image",
+    gr.Image(),
     gr.Dropdown(["yolov5s", "yolov5m"], label="Model", value="yolov5s"),
 ]
 
 outputs = [
-    "image",
+    gr.Image(elem_classes="output-image")
 ]
 
 examples = [
@@ -31,5 +31,12 @@ examples = [
     ["examples/ex4.jpg", None],
 ]
 
-iface = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, examples=examples, cache_examples=False)
+css = ".output-image {height: 700px !important}"
+
+iface = gr.Interface(fn=predict,
+                     inputs=inputs,
+                     outputs=outputs,
+                     examples=examples,
+                     cache_examples=False,
+                     css=css)
 iface.launch()