mouseland commited on
Commit
ded4361
·
verified ·
1 Parent(s): 86bfbdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -97,6 +97,7 @@ def image_resize(img, resize=400):
97
  # masks, flows, _ = model.eval(img, channels=[0,0])
98
  # return masks, flows
99
 
 
100
  def cellpose_segment(img_input):
101
  img = image_resize(img_input)
102
  #masks, flows = run_model_gpu(model, img)
 
97
  # masks, flows, _ = model.eval(img, channels=[0,0])
98
  # return masks, flows
99
 
100
+ @spaces.GPU(duration=10)
101
  def cellpose_segment(img_input):
102
  img = image_resize(img_input)
103
  #masks, flows = run_model_gpu(model, img)