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:
|