Spaces:
Running
Running
Update extract_tools.py
Browse files- 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
|
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")
|
|
|
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")
|