Update app.py
Browse files
app.py
CHANGED
@@ -334,7 +334,7 @@ class GradioWindow():
|
|
334 |
result_masks.append(masks[index])
|
335 |
return np.array(result_masks)
|
336 |
|
337 |
-
@spaces.GPU(duration=120)
|
338 |
def augment_image(self, image: Image,
|
339 |
current_object: str, new_objects_list: str,
|
340 |
ddim_steps: int, guidance_scale: int, seed: int, return_prompt: str) -> tuple:
|
|
|
334 |
result_masks.append(masks[index])
|
335 |
return np.array(result_masks)
|
336 |
|
337 |
+
# @spaces.GPU(duration=120)
|
338 |
def augment_image(self, image: Image,
|
339 |
current_object: str, new_objects_list: str,
|
340 |
ddim_steps: int, guidance_scale: int, seed: int, return_prompt: str) -> tuple:
|