AntonXue commited on
Commit
e3d2f47
1 Parent(s): 9001e40

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -8,4 +8,15 @@ datasets:
8
 
9
  This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
10
  - Library: [More Information Needed]
11
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
  This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
10
  - Library: [More Information Needed]
11
+ - Docs: [More Information Needed]
12
+
13
+ ## Usage
14
+ ```
15
+ import torch
16
+ from exlib.datasets.chestx import ChestXDataset, ChestXPathologyModel
17
+ model = ChestXPathologyModel.from_pretrained("BrachioLab/chestx_pathols")
18
+ dataset = ChestXDataset(split="test")
19
+ dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
20
+ item = next(iter(dataloader))
21
+ out = model(item["image"])
22
+ ```