Skip to content

Could not find './src/ctranspath_448_bioclinicalbert.json' and './src/ctranspath_448_pubmedbert.json' config #14

@maianhpuco

Description

@maianhpuco

in your modules/text_extractor.py

def create_text_extractor(
        model_name: str,
        device: torch.device = torch.device('cpu'),
        override_image_size = None,
):
    if model_name == 'bioclinicalbert':
        print(f'Loading {model_name} model config for text extractor.')
        with open('./src/ctranspath_448_bioclinicalbert.json') as f:
            model_cfg = json.load(f)
            
    elif model_name == 'pubmedbert':
        with open('./src/ctranspath_448_pubmedbert.json') as f:
            model_cfg = json.load(f)  ... 

This is the code for creating text extract, but i can not find the config in the ./src/ folder. Would you mind also providing this?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions