Update app.py

app.py

@@ -1,6 +1,3 @@
-import os
-import torch
-import gradio as gr
 import numpy as np
 from PIL import Image
 from einops import rearrange
@@ -9,6 +6,7 @@ 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
@@ -48,24 +46,12 @@ 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)
@@ -120,12 +106,10 @@ 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()