Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -143,7 +143,7 @@ def resize_img(input_image, max_side=768, min_side=512, size=None,
|
|
143 |
input_image = Image.fromarray(res)
|
144 |
return input_image
|
145 |
|
146 |
-
@spaces.GPU(
|
147 |
def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed, progress=gr.Progress(track_tqdm=True)):
|
148 |
|
149 |
control_mode_num = mode_mapping[control_mode]
|
|
|
143 |
input_image = Image.fromarray(res)
|
144 |
return input_image
|
145 |
|
146 |
+
@spaces.GPU()
|
147 |
def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed, progress=gr.Progress(track_tqdm=True)):
|
148 |
|
149 |
control_mode_num = mode_mapping[control_mode]
|