Update app.py

app.py

@@ -81,12 +81,9 @@ with gr.Blocks() as demo:
         gallery = gr.Image(
             label="Generated image", 
             type='pil',
-            show_label = False
-        ).style(
+            show_label = False,
             width="256",  # Use the full view width
-            height="256",  # Use the full view height
-            object_fit="scale-down"  # Maintain aspect ratio without cropping
-        )
+            height="256")  # Use the full view height
     btn.click(gr_generate_images, [text], gallery)
     gr.Examples(examples, inputs=[text])
     gr.HTML(