Skip to content

Commit f7f8b0a

Browse files
authored
docs(README): [0.1.0] Update download link
1 parent a870125 commit f7f8b0a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,11 @@
22

33
This is the repo for Bumble's *Private Detector*™ model - an image classifier that can detect lewd images.
44

5-
The internal repo has been heavily refactored and released as a fully open-source project to allow for the wider community to use and finetune a Private Detector model of their own
5+
The internal repo has been heavily refactored and released as a fully open-source project to allow for the wider community to use and finetune a Private Detector model of their own. You can download the pretrained SavedModel and checkpoint [here](https://storage.googleapis.com/private_detector/private_detector.zip)
66

77
## Model
88

9-
The SavedModel can be found in `saved_model/`, ensure you have [Git LFS](https://git-lfs.github.com/) installed to pull the full files.
9+
The SavedModel can be found in `saved_model/` within `private_detector.zip` above
1010

1111
The model is based on Efficientnet-v2 and trained on our internal dataset of lewd images - more information can be found at the whitepaper [here](pending)
1212

@@ -49,7 +49,7 @@ python3 inference.py \
4949

5050
## Additional Training
5151

52-
You can finetune the model yourself on your own data, to do so is fairly simple - though you will need the checkpoint files as can be found in `saved_checkpoint/`
52+
You can finetune the model yourself on your own data, to do so is fairly simple - though you will need the checkpoint files as can be found in `saved_checkpoint/` in `private_detector.zip`
5353

5454
Set up a JSON file with links to your image path lists for each class:
5555

0 commit comments

Comments
 (0)