Update app.py

app.py

@@ -192,18 +192,11 @@ def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_im
     
     return final_image_base64
 
-def _generate_duration(x, ckpt_name, negative_prompt, fine_edge, grow_size, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
-    try:
-        s = int(steps or 20)
-    except Exception:
-        s = 20
-    return max(60, 3 * s)
-
 
 from MagicQuill.comfy import model_management as _mm
 
 
-@spaces.GPU(duration=_generate_duration)
+@spaces.GPU(duration=20)
 def generate_image_handler(x, ckpt_name, negative_prompt, fine_edge, grow_size, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
     _mm.unload_all_models()
     if seed == -1: