mouseland commited on
Commit
465186b
·
verified ·
1 Parent(s): ab08c81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -11,6 +11,9 @@ from cellpose.io import imread, imsave
11
 
12
  # @title Data retrieval
13
  def download_weights():
 
 
 
14
  import os, requests
15
 
16
  fname = ['cpsam']
@@ -35,8 +38,8 @@ def download_weights():
35
  fid.write(r.content)
36
 
37
  try:
38
- #download_weights()
39
- model = models.CellposeModel(gpu=True) # , pretrained_model="cpsam")
40
  except Exception as e:
41
  print(f"Error loading model: {e}")
42
  exit(1)
 
11
 
12
  # @title Data retrieval
13
  def download_weights():
14
+ os.system("wget https://huggingface.co/mouseland/cellpose-sam/resolve/main/cpsam")
15
+
16
+ def download_weights_old():
17
  import os, requests
18
 
19
  fname = ['cpsam']
 
38
  fid.write(r.content)
39
 
40
  try:
41
+ download_weights()
42
+ model = models.CellposeModel(gpu=True , pretrained_model="cpsam")
43
  except Exception as e:
44
  print(f"Error loading model: {e}")
45
  exit(1)