From 2264a545e770c05e4fb24a5ca3d37be387fa0e97 Mon Sep 17 00:00:00 2001
From: Vaibhav Singh <45447817+itsvaibhav01@users.noreply.github.com>
Date: Thu, 30 May 2024 19:54:46 +0900
Subject: [PATCH] Corrected typo in get_model function

---
 libs/feature_extractor/run_clip_zero_shot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: