Update app.py

app.py

@@ -18,6 +18,16 @@ def clear_memory():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
 
+# Function to resize images while preserving the aspect ratio
+def resize_image(image, max_size=1024):
+    width, height = image.size
+    if max(width, height) > max_size:
+        ratio = max_size / max(width, height)
+        new_width = int(width * ratio)
+        new_height = int(height * ratio)
+        image = image.resize((new_width, new_height), Image.ANTIALIAS)
+    return image
+
 # Global variable definitions
 controlnet_pipe = None
 reference_pipe = None
@@ -298,6 +308,7 @@ def generate_images_with_progress(prompt, negative_prompt, batch_count, use_cont
 
         preprocessed_images = []
         for img in tqdm(control_images, desc="Preprocessing images"):
+            img = resize_image(img)  # Resize the image before preprocessing
             if controlnet_type == "Canny":
                 preprocessed_images.append(preprocess_canny(img))
             elif controlnet_type == "Depth":
@@ -356,6 +367,7 @@ def load_images_from_folder(folder_path):
         if filename.endswith(('.png', '.jpg', '.jpeg')):
             img_path = os.path.join(folder_path, filename)
             img = Image.open(img_path).convert("RGB")
+            img = resize_image(img)  # Resize the image before adding to the list
             images.append((filename, img))
     return images
 
@@ -460,7 +472,7 @@ with gr.Blocks() as demo:
             else:
                 if not batch_images_input:
                     raise ValueError("No images uploaded for batch processing.")
-                selected_images = [Image.open(img).convert("RGB") for img in batch_images_input]
+                selected_images = [resize_image(Image.open(img).convert("RGB")) for img in batch_images_input]
         else:
             selected_images = [img[1] for img in selected_folder_images]
         return generate_images_with_progress(prompt, negative_prompt, batch_count, use_controlnet, controlnet_type, mode, selected_images, progress)
@@ -486,4 +498,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()