Spaces:
Sleeping
Sleeping
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")
|