Improve rendering

app.py

@@ -82,16 +82,17 @@ def infer(
 
 
 css = """
-#col-container {
-    margin: 0 auto;
-    max-width: 1280px;
+@media (max-width: 1280px) {
+  #images-container {
+    flex-direction: column;
+  }
 }
 """
 
 with gr.Blocks(css=css) as demo:
-    with gr.Column(elem_id="col-container"):
+    with gr.Column():
         gr.Markdown("# Image-to-Image")
-        gr.Markdown(f"## Model: {model}")
+        gr.Markdown(f"## Model: `{model}`")
 
         with gr.Row():
             prompt = gr.Text(
@@ -104,7 +105,7 @@ with gr.Blocks(css=css) as demo:
 
             run_button = gr.Button("Run", scale=0, variant="primary")
 
-        with gr.Row():
+        with gr.Row(elem_id="images-container"):
             init_image = gr.Image(label="Initial image", type="pil")
 
             result = gr.Image(label="Result")