Update app.py

app.py

@@ -5,7 +5,6 @@ import numpy as np
 from PIL import Image
 from einops import rearrange
 import requests
-import spaces
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel
 from gradio_imageslider import ImageSlider
@@ -34,7 +33,6 @@ def preprocess_image(image, target_width, target_height):
     image = image.resize((target_width, target_height), Image.LANCZOS)
     return image
 
-@spaces.GPU(duration=120)
 def generate_image(prompt, control_image, control_mode, controlnet_conditioning_scale, num_steps, guidance, width, height, seed, random_seed):
     if random_seed:
         seed = np.random.randint(0, 10000)
@@ -54,7 +52,7 @@ def generate_image(prompt, control_image, control_mode, controlnet_conditioning_
         image = pipe(
             prompt,
             control_image=control_image,
-            control_mode=control_mode,
+            control_mode=control_mode,  # Pass control mode explicitly
             width=width,
             height=height,
             controlnet_conditioning_scale=controlnet_conditioning_scale,