Update app.py

app.py

@@ -1,3 +1,6 @@
+import os
+import torch
+import gradio as gr
 import numpy as np
 from PIL import Image
 from einops import rearrange
@@ -6,7 +9,6 @@ import spaces
 from huggingface_hub import login
 from gradio_imageslider import ImageSlider  # Import ImageSlider
 
-
 from image_datasets.canny_dataset import canny_processor, c_crop
 from src.flux.sampling import denoise_controlnet, get_noise, get_schedule, prepare, unpack
 from src.flux.util import load_ae, load_clip, load_t5, load_flow_model, load_controlnet, load_safetensors
@@ -46,12 +48,24 @@ def preprocess_image(image, target_width, target_height, crop=True):
         original_width, original_height = image.size
 
         # Resize to match the target size without stretching
+        scale = max(target_width / original_width, target_height / original_height)
+        resized_width = int(scale * original_width)
+        resized_height = int(scale * original_height)
+
+        image = image.resize((resized_width, resized_height), Image.LANCZOS)
+        
+        # Center crop to match the target dimensions
+        left = (resized_width - target_width) // 2
+        top = (resized_height - target_height) // 2
+        image = image.crop((left, top, left + target_width, top + target_height))
+    else:
+        image = image.resize((target_width, target_height), Image.LANCZOS)
+    
+    return image
 
 def preprocess_canny_image(image, target_width, target_height, crop=True):
     image = preprocess_image(image, target_width, target_height, crop=crop)
     image = canny_processor(image)
-
-
     return image
 
 @spaces.GPU(duration=120)
@@ -106,10 +120,12 @@ interface = gr.Interface(
         gr.Slider(minimum=128, maximum=2048, step=128, value=1024, label="Height"),
         gr.Number(value=42, label="Seed"),
         gr.Checkbox(label="Random Seed")
-
     ],
     outputs=ImageSlider(label="Before / After"),  # Use ImageSlider as the output
     title="FLUX.1 Controlnet Canny",
+    description="Generate images using ControlNet and a text prompt.\n[[non-commercial license, Flux.1 Dev](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)]"
+)
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()
+