update bug for file saving in second pass

app.py

@@ -4,6 +4,7 @@ import torch
 import gradio as gr
 import random
 from PIL import Image
+import os
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
 
 #Loading from Diffusers Library
@@ -46,7 +47,9 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
       # Open the image
       image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
       edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
-      edited_image.save(img_name, overwrite=True)
+      if os.path.exists(img_name):
+        os.remove(img_name)
+      edited_image.save(img_name) #, overwrite=True)
     else:
       seed = random.randint(0, 1000000)
       img_name = f"edited_image_{seed}.png"