diff --git a/libs/feature_extractor/run_clip_zero_shot.py b/libs/feature_extractor/run_clip_zero_shot.py index b4c559b..56545bb 100644 --- a/libs/feature_extractor/run_clip_zero_shot.py +++ b/libs/feature_extractor/run_clip_zero_shot.py @@ -19,7 +19,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") def get_model(model_name): - if modesl_name == const.CLIP_BASE_NAME: + if model_name == const.CLIP_BASE_NAME: clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32") clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32") elif model_name == const.CLIP_ALIGN_NAME: