OriLib commited on
Commit
aab1fdd
1 Parent(s): 4cdcbb4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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")