Update README.md
Browse files
README.md
CHANGED
@@ -104,8 +104,8 @@ from PIL import Image
|
|
104 |
from briarmbg import BriaRMBG
|
105 |
from utilities import preprocess_image, postprocess_image
|
106 |
|
107 |
-
model_path = f"{os.path.dirname(__file__)}/model.pth"
|
108 |
-
im_path = f"{os.path.dirname(__file__)}/example_input.jpg"
|
109 |
|
110 |
net = BriaRMBG()
|
111 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
104 |
from briarmbg import BriaRMBG
|
105 |
from utilities import preprocess_image, postprocess_image
|
106 |
|
107 |
+
model_path = f"{os.path.dirname(os.path.abspath(__file__))}/model.pth"
|
108 |
+
im_path = f"{os.path.dirname(os.path.abspath(__file__))}/example_input.jpg"
|
109 |
|
110 |
net = BriaRMBG()
|
111 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|