Update app.py

app.py

@@ -56,7 +56,19 @@ examples = [
 css = """
 footer {visibility: hidden;}
 .container {max-width: 1200px; margin: auto; padding: 20px;}
-.generate-box {background-color: #f0f0f0; border-radius: 10px; padding: 20px; margin-bottom: 20px;}
+.generate-box, .image-box {
+    background-color: #f0f0f0;
+    border-radius: 10px;
+    padding: 20px;
+    margin-bottom: 20px;
+    height: 100%;  /* Set height to 100% */
+    display: flex;
+    flex-direction: column;
+}
+.image-box img {
+    max-height: 100%;
+    object-fit: contain;
+}
 .generate-box .row {display: flex; align-items: center; margin-bottom: 10px;}
 .generate-box .row > * {margin-right: 10px;}
 .generate-box .row > *:last-child {margin-right: 0;}
@@ -71,7 +83,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
         with gr.Row():
             with gr.Column(scale=2):
-                result = gr.Image(label="Generated Image", show_label=False, interactive=False)
+                result = gr.Image(label="Generated Image", show_label=False, interactive=False, elem_classes="image-box")
             with gr.Column(scale=1):
                 with gr.Column(elem_classes="generate-box"):
                     prompt = gr.Text(
@@ -94,7 +106,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                         height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
 
         with gr.Column(elem_classes="examples-gallery"):
-            gr.Markdown("### 🌟 Inspiration Gallery")
+            gr.Markdown("### Gallery")
             gr.Examples(
                 examples=examples,
                 fn=generate_image,