Spaces:

pengpeng2000
/

sam

Runtime error

sam / sam.py

pengpeng2000

Create sam.py

0032037 verified over 2 years ago

232 Bytes

# Load model directly
from transformers import AutoProcessor, AutoModelForMaskGeneration

processor = AutoProcessor.from_pretrained("facebook/sam-vit-base")
model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-base")

	# Load model directly
	from transformers import AutoProcessor, AutoModelForMaskGeneration

	processor = AutoProcessor.from_pretrained("facebook/sam-vit-base")
	model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-base")