Hugging Face's logo

Spaces:
LXT
/
OMG_Seg
Sleeping

App Files Community
4
HarborYuan commited on
Commit
d6c0d66
·
verified ·
1 Parent(s): 71a4a35

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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]