lelloman commited on
Commit
4f4cd5d
·
verified ·
1 Parent(s): 824ef96

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +58 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ tags:
6
+ - text-classification
7
+ - onnx
8
+ - bert
9
+ - torrent
10
+ - content-classification
11
+ base_model: prajjwal1/bert-tiny
12
+ pipeline_tag: text-classification
13
+ ---
14
+
15
+ # BERT Torrent Classifier
16
+
17
+ A fine-tuned BERT-tiny model for classifying torrent content into media types.
18
+
19
+ ## Model Details
20
+
21
+ - **Base model:** [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny)
22
+ - **Task:** Multi-class text classification
23
+ - **Labels:** audio, video, software, book, other
24
+ - **Format:** ONNX (with embedded weights)
25
+ - **Size:** ~17MB
26
+
27
+ ## Training
28
+
29
+ - **Training data:** ~10k torrent names with 4-LLM consensus voting
30
+ - **LLM ensemble:** qwen2.5:3b, gemma3:4b, mistral:7b, qwen3-coder:30b
31
+ - **Consensus rules:** 4-agree = high confidence, 3v1 = majority vote, 2v2 = discarded
32
+ - **Accuracy:** ~92% on held-out test set
33
+
34
+ ## Usage
35
+
36
+ This model is designed for use with [mimmo](https://github.com/lelloman/mimmo), a Rust library for torrent content classification. The ONNX model is embedded directly in the binary at compile time.
37
+
38
+ ```rust
39
+ // Model is automatically downloaded during build
40
+ const MODEL_BYTES: &[u8] = include_bytes!("../models/bert/model_embedded.onnx");
41
+ const TOKENIZER_JSON: &str = include_str!("../models/bert/tokenizer.json");
42
+ ```
43
+
44
+ ## Performance
45
+
46
+ - Inference: <10ms per sample (CPU)
47
+ - Used as ML fallback when pattern matching is inconclusive
48
+
49
+ ## Files
50
+
51
+ - `model_embedded.onnx` - ONNX model with embedded weights
52
+ - `tokenizer.json` - HuggingFace tokenizer
53
+ - `vocab.txt` - Vocabulary file
54
+ - `config.json` - Model configuration
55
+
56
+ ## License
57
+
58
+ MIT