Update README.md

README.md

@@ -6,4 +6,15 @@ tags:
 
 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]
+- Docs: [More Information Needed]
+
+## Usage
+```
+import torch
+from exlib.datasets.cholec import CholecDataset, CholecModel
+model = CholecModel.from_pretrained("BrachioLab/cholecystectomy_organs")
+dataset = CholecDataset(split="test")
+dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
+item = next(iter(dataloader))
+out = model(item["image"])
+```