Update app.py

app.py

@@ -106,9 +106,21 @@ with gr.Blocks() as demo:
 
             btn = gr.Button("Generate image").style(full_width=False)
 
-        gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(columns=[1], rows=[1], object_fit="scale-down", height="50%", width="50%")
+        #gallery = gr.Gallery(
+        #    label="Generated images", show_label=False, elem_id="gallery"
+        #).style(columns=[1], rows=[1], object_fit="scale-down", height="50%", width="50%")
+
+
+        gallery = gr.Image(
+            label="Generated image", 
+            show_label = False,
+            tool=None,  # Disable editing tools if not needed
+            source="upload")  # Allows image uploading
+            .style(
+            width="50%",  # Use the full view width
+            height="50%",  # Use the full view height
+            object_fit="scale-down"  # Maintain aspect ratio without cropping
+        )
 
     num_images_slider = 1
     num_inference_steps_slider = 20