Instructions to use aradootle/sam-vit-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aradootle/sam-vit-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("mask-generation", model="aradootle/sam-vit-base")# Load model directly from transformers import AutoProcessor, AutoModelForMaskGeneration processor = AutoProcessor.from_pretrained("aradootle/sam-vit-base") model = AutoModelForMaskGeneration.from_pretrained("aradootle/sam-vit-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| from handler import EndpointHandler | |
| # init handler | |
| my_handler = EndpointHandler(path=".") | |
| # prepare sample payload | |
| payload = {"inputs": "I am quite excited how this will turn out", "imageUrl": "https://res.cloudinary.com/dvfgdnfzd/image/upload/v1693510414/nvae1t0lvgzavfkgb45j.png"} | |
| # test the handler | |
| payload=my_handler(payload) | |
| # show results | |
| print("payload", payload) |