TheoBH commited on
Commit
4e06e61
·
verified ·
1 Parent(s): 0d55c0d

Update predict.py

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -48,7 +48,7 @@ def predict_masks(input_img_path: str):
48
 
49
  #load model and image processor
50
  default_ckpt = load_default_ckpt()
51
- model, image_processor = load_model_and_processor()
52
 
53
  ## pass input image through image processor
54
  image = Image.open(input_img_path)
 
48
 
49
  #load model and image processor
50
  default_ckpt = load_default_ckpt()
51
+ model, image_processor = load_model_and_processor(default_ckpt)
52
 
53
  ## pass input image through image processor
54
  image = Image.open(input_img_path)