not-lain commited on
Commit
f559c9b
1 Parent(s): 43f9004

update readme

Browse files
Files changed (1) hide show
  1. README.md +13 -1
README.md CHANGED
@@ -8,4 +8,16 @@ tags:
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: https://github.com/SkyTNT/anime-segmentation
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: https://github.com/SkyTNT/anime-segmentation
11
+
12
+ ## How to load the model
13
+ ```
14
+ wget https://raw.githubusercontent.com/SkyTNT/anime-segmentation/refs/heads/main/train.py
15
+ ```
16
+ ```
17
+ pip install pytorch_lightning kornia timm "huggingface_hub>=0.22"
18
+ ```
19
+
20
+ ```python
21
+ from train import AnimeSegmentation
22
+ model = AnimeSegmentation.from_pretrained("skytnt/anime-seg")
23
+ ```