fantos commited on
Commit
5221f40
·
verified ·
1 Parent(s): 14fe83c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(duration=180)
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]