Update app.py

app.py

@@ -1,5 +1,8 @@
 from __future__ import annotations
 
+import math
+import random
+
 import gradio as gr
 import torch
 from PIL import Image, ImageOps
@@ -64,7 +67,7 @@ def main():
 
         generator = torch.manual_seed(seed)
         edited_image = pipe(
-            instruction, 
+            instruction,
             guidance_scale=text_cfg_scale, image_guidance_scale=image_cfg_scale,
             num_inference_steps=steps, generator=generator,
         ).images[0]
@@ -86,7 +89,11 @@ def main():
                 reset_button = gr.Button("Reset")
             with gr.Column(scale=3):
                 instruction = gr.Textbox(lines=1, label="Edit Instruction", interactive=True)
-
+        with gr.Row():
+            
+            edited_image = gr.Image(label=f"Edited Image", type="pil", interactive=False)
+         
+            edited_image.style(height=512, width=512)
         with gr.Row():
             steps = gr.Number(value=50, precision=0, label="Steps", interactive=True)
             randomize_seed = gr.Radio(