andreped commited on
Commit
82a07e9
·
1 Parent(s): acc7ac7
Files changed (1) hide show
  1. livermask/livermask.py +2 -1
livermask/livermask.py CHANGED
@@ -40,7 +40,7 @@ def post_process(pred):
40
 
41
  def get_model():
42
  url = "https://drive.google.com/uc?id=12or5Q79at2BtLgQ7IaglNGPFGRlEgEHc"
43
- output = "model.h5"
44
  md5 = "ef5a6dfb794b39bea03f5496a9a49d4d"
45
  gdown.cached_download(url, output, md5=md5) #, postprocess=gdown.extractall)
46
 
@@ -48,6 +48,7 @@ def func(path, output, cpu):
48
 
49
  cwd = "/".join(os.path.realpath(__file__).replace("\\", "/").split("/")[:-1]) + "/"
50
 
 
51
  name = cwd + "model.h5"
52
 
53
  # get model
 
40
 
41
  def get_model():
42
  url = "https://drive.google.com/uc?id=12or5Q79at2BtLgQ7IaglNGPFGRlEgEHc"
43
+ output = "./model.h5"
44
  md5 = "ef5a6dfb794b39bea03f5496a9a49d4d"
45
  gdown.cached_download(url, output, md5=md5) #, postprocess=gdown.extractall)
46
 
 
48
 
49
  cwd = "/".join(os.path.realpath(__file__).replace("\\", "/").split("/")[:-1]) + "/"
50
 
51
+ print("Current cwd:", cwd)
52
  name = cwd + "model.h5"
53
 
54
  # get model