mouseland commited on
Commit
0313286
·
verified ·
1 Parent(s): 1c850b8

Update app.py

Browse files

Does ZeroSpace work now?

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -90,6 +90,7 @@ def image_resize(img, resize=400):
90
  img = img.astype(np.uint8)
91
  return img
92
 
 
93
  def cellpose_segment(img_input):
94
  img = image_resize(img_input)
95
  masks, flows, _ = model.eval(img, channels=[0,0])
 
90
  img = img.astype(np.uint8)
91
  return img
92
 
93
+ @spaces.GPU
94
  def cellpose_segment(img_input):
95
  img = image_resize(img_input)
96
  masks, flows, _ = model.eval(img, channels=[0,0])