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,
|