Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 import os
 import random
+import space
 import numpy as np
 import cv2
 from PIL import Image
@@ -127,6 +128,7 @@ def get_control_image(input_image, control_mode):
 # --- 4. Main Generation Function ---
 MAX_SEED = np.iinfo(np.int32).max
 
+@space.GPU(duration=120)
 def generate(
     image,
     prompt,