Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -128,7 +128,7 @@ def get_control_image(input_image, control_mode):
|
|
128 |
# --- 4. Main Generation Function ---
|
129 |
MAX_SEED = np.iinfo(np.int32).max
|
130 |
|
131 |
-
@
|
132 |
def generate(
|
133 |
image,
|
134 |
prompt,
|
|
|
128 |
# --- 4. Main Generation Function ---
|
129 |
MAX_SEED = np.iinfo(np.int32).max
|
130 |
|
131 |
+
@spaces.GPU(duration=120)
|
132 |
def generate(
|
133 |
image,
|
134 |
prompt,
|