update code to handle bug in image save

app.py

@@ -38,16 +38,13 @@ be to the input. This pipeline requires a value of at least `1`. It's possible y
     * Cropping the image so the face takes up a larger portion of the frame.
 """
 
-def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, counter_out, prompt, history, progress=gr.Progress(track_tqdm=True)):
-    progress(0, desc="Starting...")
-    #global counter 
-    #counter += 1
+def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, counter_out, prompt, history): #, progress=gr.Progress(track_tqdm=True)):
     #if message == "revert": --to add revert functionality later
     print(f"counter:{counter_out}, prompt:{prompt}, img_name:{img_name}")
     if counter_out > 0:
       # 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]
+      #image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
+      edited_image = pipe(prompt, image=image_hid, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
       if os.path.exists(img_name):
         os.remove(img_name)
       edited_image.save(img_name) #, overwrite=True)
@@ -81,7 +78,7 @@ with gr.Blocks() as demo:
           in_steps = gr.Number(label="Enter the number of Inference steps", value = 20)
           in_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Guidance scale", value=7.5)
           in_img_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Image Guidance scale", value=1.5)
-          image_hid = gr.Image(visible=False,)
+          image_hid = gr.Image(visible=False,type='pil')
           img_name_temp_out = gr.Textbox(visible=False)
           counter_out = gr.Number(visible=False, value=0)
       chatbot = gr.Chatbot()