# pip install -U transformers accelerate
# Load model directly
from transformers import AutoProcessor, AutoModelForMaskGeneration
processor = AutoProcessor.from_pretrained("Mit1208/sam-fine-tune-doclaynet")
model = AutoModelForMaskGeneration.from_pretrained("Mit1208/sam-fine-tune-doclaynet", device_map="auto")Quick Links
I have Fine tuned https://huggingface.co/facebook/sam-vit-base using Doclaynet-small dataset.
My focus was to detect structure from the documents like paragraph, header, tabel using SOTA model Segment Anything Model (SAM).
- Downloads last month
- 14
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("mask-generation", model="Mit1208/sam-fine-tune-doclaynet")