mouseland commited on
Commit
644bb2c
·
verified ·
1 Parent(s): 180fb05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def download_weights():
32
 
33
  try:
34
  #download_weights()
35
- model = models.CellposeModel(gpu=True, pretrained_model="cyto3", channels = [0,0])
36
  except Exception as e:
37
  print(f"Error loading model: {e}")
38
  exit(1)
@@ -127,7 +127,7 @@ def image_resize(img, resize=400):
127
 
128
  @spaces.GPU(duration=10)
129
  def run_model_gpu(img):
130
- masks, flows, _ = model.eval(img)
131
  return masks, flows
132
 
133
  #@spaces.GPU(duration=10)
 
32
 
33
  try:
34
  #download_weights()
35
+ model = models.CellposeModel(gpu=True, pretrained_model="cyto3")
36
  except Exception as e:
37
  print(f"Error loading model: {e}")
38
  exit(1)
 
127
 
128
  @spaces.GPU(duration=10)
129
  def run_model_gpu(img):
130
+ masks, flows, _ = model.eval(img, channels = [0,0])
131
  return masks, flows
132
 
133
  #@spaces.GPU(duration=10)