Spaces:
Sleeping
Sleeping
Commit
•
d6c0d66
1
Parent(s):
71a4a35
Update main.py
Browse files
main.py
CHANGED
|
@@ -109,7 +109,7 @@ def get_points_with_draw(image, img_state, evt: gr.SelectData):
|
|
| 109 |
return image
|
| 110 |
|
| 111 |
|
| 112 |
-
@spaces.GPU
|
| 113 |
def segment_point(image, img_state, mode):
|
| 114 |
output_img = img_state.img
|
| 115 |
h, w = output_img.shape[:2]
|
|
|
|
| 109 |
return image
|
| 110 |
|
| 111 |
|
| 112 |
+
@spaces.GPU()
|
| 113 |
def segment_point(image, img_state, mode):
|
| 114 |
output_img = img_state.img
|
| 115 |
h, w = output_img.shape[:2]
|