Spaces:
Running
on
L40S
Running
on
L40S
Update segment.py
Browse files- segment.py +1 -1
segment.py
CHANGED
|
@@ -98,7 +98,7 @@ def draw_panoptic_segmentation(segmentation, segments_info,save_folder=None, nos
|
|
| 98 |
return mask_np_list,label_list
|
| 99 |
|
| 100 |
|
| 101 |
-
|
| 102 |
def run_segmentation(image, name="example_tmp", size = 512, noseg=False):
|
| 103 |
|
| 104 |
base_folder_path = "."
|
|
|
|
| 98 |
return mask_np_list,label_list
|
| 99 |
|
| 100 |
|
| 101 |
+
@spaces.GPU(duration=10)
|
| 102 |
def run_segmentation(image, name="example_tmp", size = 512, noseg=False):
|
| 103 |
|
| 104 |
base_folder_path = "."
|