Merge pull request #17 from healthonrails/patch-1

Fix typo
luca-medeiros · May 27, 2023 · 32cfe0f · 32cfe0f
2 parents bde5ef7 + 31dfad4
commit 32cfe0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lang_sam/lang_sam.py b/lang_sam/lang_sam.py
@@ -70,9 +70,9 @@ def build_sam(self, sam_type):
 
  def build_groundingdino(self):
  ckpt_repo_id = "ShilongLiu/GroundingDINO"
- ckpt_filenmae = "groundingdino_swinb_cogcoor.pth"
+ ckpt_filename = "groundingdino_swinb_cogcoor.pth"
  ckpt_config_filename = "GroundingDINO_SwinB.cfg.py"
- self.groundingdino = load_model_hf(ckpt_repo_id, ckpt_filenmae, ckpt_config_filename)
+ self.groundingdino = load_model_hf(ckpt_repo_id, ckpt_filename, ckpt_config_filename)
 
  def predict_dino(self, image_pil, text_prompt, box_threshold, text_threshold):
  image_trans = transform_image(image_pil)