Update app.py

app.py

@@ -37,9 +37,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("""<h1 style="text-align: center;">Synthetic Image Generator</h1>""")
     
     with gr.Tab("Generate Single Image"):
-      gr.Markdown("## Using the Synthetic Generator\n\nSimply click 'Generate Image' to create a synthetic image. The generated image will be displayed below.")
-      gen_img = gr.Image( tool="select", type="numpy", label="Generated Image").style(height=256, width=256, rounded=True)
-      gen_button = gr.Button("Generate", variant="primary")
+      with gr.Row():
+        with gr.Column():
+          gr.Markdown("""## Using the Synthetic Image Generator\n\nSimply click 'Generate Image' to create a synthetic image.\n""")
+          gr.Image("./train.png",label="Training Image sample").style( rounded=True, scale=1)
+          gen_button = gr.Button("Generate", variant="primary")
+        gen_img = gr.Image( tool="select", type="numpy", label="Generated Image").style(height=512, width=512, rounded=True)
 
     with gr.Tab("Generate Multiple Images"):
       gr.Markdown(