BrachioLab/cholecystectomy
Viewer • Updated • 1.02k • 19
How to use BrachioLab/cholecystectomy_gonogo with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("BrachioLab/cholecystectomy_gonogo", dtype="auto")This model has been pushed to the Hub using the PytorchModelHubMixin integration:
import torch
from exlib.datasets.cholec import CholecDataset, CholecModel
model = CholecModel.from_pretrained("BrachioLab/cholecystectomy_gonogo")
dataset = CholecDataset(split="test")
dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
item = next(iter(dataloader))
out = model(item["image"])