Samarth991 commited on
Commit
843e5f7
·
verified ·
1 Parent(s): 4708376

Update extract_tools.py

Browse files
Files changed (1) hide show
  1. extract_tools.py +2 -2
extract_tools.py CHANGED
@@ -31,12 +31,12 @@ try:
31
  maskformer_processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
32
  maskformer_model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
33
  except:
34
- logging.error("Unable to Maskformer model {}".format(err))
35
 
36
  try:
37
  yolo_world_model= YoloWorld()
38
  except :
39
- logging.error("Unable to Yolo world model {}".format(err))
40
 
41
  def get_groq_model(model_name = "gemma2-9b-it"):
42
  os.environ.get("GROQ_API_KEY")
 
31
  maskformer_processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
32
  maskformer_model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
33
  except:
34
+ logging.error("Unable to Mask2Former model")
35
 
36
  try:
37
  yolo_world_model= YoloWorld()
38
  except :
39
+ logging.error("Unable to Yolo world model")
40
 
41
  def get_groq_model(model_name = "gemma2-9b-it"):
42
  os.environ.get("GROQ_API_KEY")