Spaces:
Running
Running
Update predict.py
Browse files- 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)
|