Spaces:
azhan77168
/
Running on Zero

azhan77168 commited on
Commit
c0c664f
·
verified ·
1 Parent(s): 37505a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -210,7 +210,7 @@ def guess_prompt_handler(original_image, add_color_image, add_edge_image):
210
 
211
  return res
212
 
213
- @spaces.GPU(duration=120)
214
  def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_image, remove_edge_image, positive_prompt, negative_prompt, grow_size, stroke_as_edge, fine_edge, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
215
  add_color_image, original_image, total_mask, add_edge_mask, remove_edge_mask = prepare_images_and_masks(total_mask, original_image, add_color_image, add_edge_image, remove_edge_image)
216
  progress = None
 
210
 
211
  return res
212
 
213
+ @spaces.GPU
214
  def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_image, remove_edge_image, positive_prompt, negative_prompt, grow_size, stroke_as_edge, fine_edge, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
215
  add_color_image, original_image, total_mask, add_edge_mask, remove_edge_mask = prepare_images_and_masks(total_mask, original_image, add_color_image, add_edge_image, remove_edge_image)
216
  progress = None