Update my_model/utilities/gen_utilities.py
Browse files
    	
        my_model/utilities/gen_utilities.py
    CHANGED
    
    | @@ -275,26 +275,27 @@ def get_image_path(name, path_type): | |
| 275 |  | 
| 276 | 
             
            def get_model_path(model_name):
         | 
| 277 | 
             
                """
         | 
| 278 | 
            -
                Get the path to the specified model  | 
| 279 |  | 
| 280 | 
             
                Args:
         | 
| 281 | 
            -
                    model_name (str):  | 
| 282 |  | 
| 283 | 
             
                Returns:
         | 
| 284 | 
            -
                    str: Absolute path to the specified model  | 
| 285 | 
             
                """
         | 
| 286 | 
             
                # Directory of the current script
         | 
| 287 | 
             
                current_script_dir = os.path.dirname(os.path.abspath(__file__))
         | 
| 288 |  | 
| 289 | 
            -
                # Directory of the ' | 
| 290 | 
            -
                 | 
| 291 |  | 
| 292 | 
             
                # Path to the 'models/{model_name}' folder
         | 
| 293 | 
            -
                model_path = os.path.join( | 
| 294 |  | 
| 295 | 
             
                return model_path
         | 
| 296 |  | 
| 297 |  | 
|  | |
| 298 | 
             
            def free_gpu_resources():
         | 
| 299 | 
             
                """
         | 
| 300 | 
             
                Clears GPU memory.
         | 
|  | |
| 275 |  | 
| 276 | 
             
            def get_model_path(model_name):
         | 
| 277 | 
             
                """
         | 
| 278 | 
            +
                Get the path to the specified model folder.
         | 
| 279 |  | 
| 280 | 
             
                Args:
         | 
| 281 | 
            +
                    model_name (str): Name of the model folder.
         | 
| 282 |  | 
| 283 | 
             
                Returns:
         | 
| 284 | 
            +
                    str: Absolute path to the specified model folder.
         | 
| 285 | 
             
                """
         | 
| 286 | 
             
                # Directory of the current script
         | 
| 287 | 
             
                current_script_dir = os.path.dirname(os.path.abspath(__file__))
         | 
| 288 |  | 
| 289 | 
            +
                # Directory of the 'app' folder (parent of the 'my_model' folder)
         | 
| 290 | 
            +
                app_dir = os.path.dirname(os.path.dirname(current_script_dir))
         | 
| 291 |  | 
| 292 | 
             
                # Path to the 'models/{model_name}' folder
         | 
| 293 | 
            +
                model_path = os.path.join(app_dir, "models", model_name)
         | 
| 294 |  | 
| 295 | 
             
                return model_path
         | 
| 296 |  | 
| 297 |  | 
| 298 | 
            +
             | 
| 299 | 
             
            def free_gpu_resources():
         | 
| 300 | 
             
                """
         | 
| 301 | 
             
                Clears GPU memory.
         | 
