Update app.py

app.py

@@ -44,7 +44,6 @@ example_imgs = [
     ["res/example_pair2.jpg", 0.4]
 ]
 
-css = ".output-image, .input-image, .image-preview {height: 600px !important}"
-interface = gr.Interface(fn=predict, inputs=[gr.Image(type="pil", label="Screenshot"), gr.Slider(0.0, 1.0, step=0.1, value=0.4)], outputs=gr.Image(type="pil", label="Annotated Screenshot"), examples=example_imgs, css=css)
+interface = gr.Interface(fn=predict, inputs=[gr.Image(type="pil", label="Screenshot"), gr.Slider(0.0, 1.0, step=0.1, value=0.4)], outputs=gr.Image(type="pil", label="Annotated Screenshot").style(height=600), examples=example_imgs)
 
 interface.launch()