mouseland commited on
Commit
16ebf6e
·
verified ·
1 Parent(s): 64c85bd

Update app.py

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