Update app.py

app.py

@@ -5,6 +5,7 @@ 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
@@ -32,7 +33,8 @@ CONTROL_MODES = {
 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)