Update app.py
Browse files
app.py
CHANGED
|
@@ -150,25 +150,25 @@ def image_resize(img, resize=400):
|
|
| 150 |
|
| 151 |
@spaces.GPU(duration=10)
|
| 152 |
def run_model_gpu(img, max_iter):
|
|
|
|
| 153 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 154 |
return masks, flows
|
| 155 |
|
| 156 |
@spaces.GPU(duration=60)
|
| 157 |
def run_model_gpu60(img, max_iter):
|
|
|
|
| 158 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 159 |
return masks, flows
|
| 160 |
|
| 161 |
@spaces.GPU(duration=240)
|
| 162 |
def run_model_gpu240(img, max_iter):
|
|
|
|
| 163 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 164 |
return masks, flows
|
| 165 |
|
| 166 |
-
|
| 167 |
-
|
| 168 |
from zipfile import ZipFile
|
| 169 |
def cellpose_segment(filepath, resize = 1000,max_iter = 250):
|
| 170 |
|
| 171 |
-
print(max_iter)
|
| 172 |
zip_path = os.path.splitext(filepath[-1])[0]+"_masks.zip"
|
| 173 |
#zip_path = 'masks.zip'
|
| 174 |
with ZipFile(zip_path, 'w') as myzip:
|
|
|
|
| 150 |
|
| 151 |
@spaces.GPU(duration=10)
|
| 152 |
def run_model_gpu(img, max_iter):
|
| 153 |
+
print(max_iter)
|
| 154 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 155 |
return masks, flows
|
| 156 |
|
| 157 |
@spaces.GPU(duration=60)
|
| 158 |
def run_model_gpu60(img, max_iter):
|
| 159 |
+
print(max_iter)
|
| 160 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 161 |
return masks, flows
|
| 162 |
|
| 163 |
@spaces.GPU(duration=240)
|
| 164 |
def run_model_gpu240(img, max_iter):
|
| 165 |
+
print(max_iter)
|
| 166 |
masks, flows, _ = model.eval(img, niter = max_iter)#, channels = [0,0])
|
| 167 |
return masks, flows
|
| 168 |
|
|
|
|
|
|
|
| 169 |
from zipfile import ZipFile
|
| 170 |
def cellpose_segment(filepath, resize = 1000,max_iter = 250):
|
| 171 |
|
|
|
|
| 172 |
zip_path = os.path.splitext(filepath[-1])[0]+"_masks.zip"
|
| 173 |
#zip_path = 'masks.zip'
|
| 174 |
with ZipFile(zip_path, 'w') as myzip:
|