lodestones
/

tagger-experiment

Image Classification

multi-label-classification

Model card Files Files and versions

lodestones commited on 18 days ago

Commit

f6153b6

·

verified ·

1 Parent(s): 79e5308

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -53,7 +53,7 @@ backbone fine-tuned end-to-end with a single linear projection head.
 from inference_tagger_standalone import Tagger
 tagger = Tagger(
-    checkpoint_path="2026-03-28_22-57-47.safetensors",
     vocab_path="tagger_vocab.json",
     device="cuda",
 )
@@ -70,7 +70,7 @@ tags = tagger.predict("https://example.com/image.jpg", threshold=0.35)
 ```bash
 # top-30 tags, pretty output
 python inference_tagger_standalone.py \
-    --checkpoint 2026-03-28_22-57-47.safetensors \
     --vocab tagger_vocab.json \
     --images photo.jpg https://example.com/image.jpg \
     --topk 30
@@ -88,7 +88,7 @@ python inference_tagger_standalone.py ... --format json
 pip install fastapi uvicorn jinja2 aiofiles
 python tagger_ui_server.py \
-    --checkpoint 2026-03-28_22-57-47.safetensors \
     --vocab tagger_vocab.json \
     --port 7860
 # → open http://localhost:7860

 from inference_tagger_standalone import Tagger
 tagger = Tagger(
+    checkpoint_path="tagger_proto.safetensors",
     vocab_path="tagger_vocab.json",
     device="cuda",
 )
 ```bash
 # top-30 tags, pretty output
 python inference_tagger_standalone.py \
+    --checkpoint tagger_proto.safetensors \
     --vocab tagger_vocab.json \
     --images photo.jpg https://example.com/image.jpg \
     --topk 30
 pip install fastapi uvicorn jinja2 aiofiles
 python tagger_ui_server.py \
+    --checkpoint tagger_proto.safetensors \
     --vocab tagger_vocab.json \
     --port 7860
 # → open http://localhost:7860