Update app.py

app.py

@@ -311,7 +311,7 @@ def process_image_batch(images, pipe, prompt, negative_prompt, num_inference_ste
         clear_memory()  # Clear memory after each batch
     return all_processed_images
 
-# Function to generate images
+# Define the function to generate images
 def generate_images_with_progress(prompt, negative_prompt, batch_count, use_controlnet, controlnet_type, mode, control_images, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
     global controlnet_pipe, pipe, reference_pipe
 
@@ -378,13 +378,8 @@ def generate_images_with_progress(prompt, negative_prompt, batch_count, use_cont
 
     clear_memory()
 
-    # Ensure all generated images are in PIL format and save as PNG
-    pil_images = []
-    for img in images:
-        if not isinstance(img, Image.Image):
-            img = Image.fromarray(np.array(img))
-        img.save("temp.png", format="PNG")
-        pil_images.append(Image.open("temp.png"))
+    # Ensure all generated images are in PIL format
+    pil_images = [img if isinstance(img, Image.Image) else Image.fromarray(np.array(img)) for img in images]
 
     return pil_images
 
@@ -563,4 +558,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     # Your Gradio interface setup here
     demo.launch(auth=("roland", "roland"), debug=True)
-    clear_memory()
+    clear_memory()