Update app.py
Browse filesDoes ZeroSpace work now?
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])
|