Update app.py

app.py

@@ -192,8 +192,9 @@ with gr.Blocks(title=TITLE, css=custom_css) as app:
                     label="Original vs. Colorized",
                     columns=2,
                     elem_id="comparison_gallery",
-                    height="auto"
-                ).style(grid=[2], height="auto")
+                    height="auto",
+                    grid=[2]
+                )
                 download_btn = gr.File(
                     label="Download Colorized Image",
                     elem_id="download-btn"
@@ -214,4 +215,4 @@ with gr.Blocks(title=TITLE, css=custom_css) as app:
 # Launch app
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
-    app.queue().launch(server_name="0.0.0.0", server_port=port)
+    app.queue().launch(server_name="0.0.0.0", server_port=port)