How to use riversnow/so101-segmentation-model with ultralytics:
# Couldn't find a valid YOLO version tag. # Replace XX with the correct version. from ultralytics import YOLOvXX model = YOLOvXX.from_pretrained("riversnow/so101-segmentation-model") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)