--- tags: - pytorch_model_hub_mixin - model_hub_mixin datasets: - BrachioLab/cholecystectomy --- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: - Library: [More Information Needed] - Docs: [More Information Needed] ## Usage ``` 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"]) ```