Update app.py

app.py

@@ -37,7 +37,7 @@ def dilate_mask(mask, kernel_size=5, iterations=5):
     mask = cv2.dilate(np.array(mask), kernel, iterations=iterations)
     return Image.fromarray(mask)
 
-@spaces.GPU
+@spaces.GPU(duration=20)
 def remove_obj(image, uploaded_mask, seed):
     image_pil = image["image"].resize((512, 512), Image.ANTIALIAS)
     mask = dilate_mask(uploaded_mask)