Update example_inference.py
Browse files- example_inference.py +2 -2
    	
        example_inference.py
    CHANGED
    
    | @@ -6,8 +6,8 @@ from utilities import preprocess_image, postprocess_image | |
| 6 |  | 
| 7 | 
             
            def example_inference():
         | 
| 8 |  | 
| 9 | 
            -
                model_path = f"{os.path.dirname(__file__)}/model.pth"
         | 
| 10 | 
            -
                im_path = f"{os.path.dirname(__file__)}/example_input.jpg"
         | 
| 11 |  | 
| 12 | 
             
                net = BriaRMBG()
         | 
| 13 | 
             
                device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         | 
|  | |
| 6 |  | 
| 7 | 
             
            def example_inference():
         | 
| 8 |  | 
| 9 | 
            +
                model_path = f"{os.path.dirname(os.path.abspath(__file__))}/model.pth"
         | 
| 10 | 
            +
                im_path = f"{os.path.dirname(os.path.abspath(__file__))}/example_input.jpg"
         | 
| 11 |  | 
| 12 | 
             
                net = BriaRMBG()
         | 
| 13 | 
             
                device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         | 

