update

README.md

@@ -13,6 +13,17 @@ This model uses the **ThaoNet-Small** architecture, optimized for speed and low
 | **Input Size** | `32px` | Lower resolution for speed. |
 | **Params** | **~1.6 Million** | Very small, suitable for mobile/CPU. |
 
+## File Structure
+```
+model9k/
+├── model.safetensors      # PyTorch weights (SafeTensors format)
+├── model.onnx             # Exported ONNX model
+├── config.yml             # Model configuration
+├── khmer_dict.txt         # Character vocabulary list
+├── model_vocab.json       # Full vocabulary mapping
+└── README.md              # This file
+```
+
 ## Usage
 
 ### 1. Run Inference (ONNX)
@@ -26,7 +37,15 @@ python tools/export/predict.py \
 ```
 *Note: Ensure you use `--height 32` as this model was trained on lower resolution images.*
 
-### 2. Performance & Data
+### 2. Load Weights (SafeTensors)
+
+```python
+from safetensors.torch import load_file
+state_dict = load_file("model9k/model.safetensors")
+# load into model...
+```
+
+### 3. Performance & Data
 *   **Training Data**: 9,000 (9k) synthetic Khmer text line images.
 *   **CER (Character Error Rate)**: ~15-20% (Estimated on diverse data).
 *   **WER (Word Error Rate)**: ~30-40%.

config.yml

@@ -0,0 +1,12 @@
+model:
+  name: small
+  target_h: 32
+  backbone:
+    type: lightweight
+  head:
+    type: transformer_ctc
+    d_model: 128
+    num_layers: 2
+vocab:
+  blank_token: '[BLANK]'
+  pad_token: '[PAD]'

khmer_dict.txt

@@ -0,0 +1,179 @@
+A
+B
+C
+D
+E
+F
+G
+H
+I
+J
+K
+L
+M
+N
+O
+P
+Q
+R
+S
+T
+U
+V
+W
+X
+Y
+Z
+a
+b
+c
+d
+e
+f
+g
+h
+i
+j
+k
+l
+m
+n
+o
+p
+q
+r
+s
+t
+u
+v
+w
+x
+y
+z
+0
+1
+2
+3
+4
+5
+6
+7
+8
+9
+.
+,
+!
+?
+(
+)
+[
+]
+-
++
+/
+:
+;
+"
+'
+ 
+ក
+ខ
+គ
+ឃ
+ង
+ច
+ឆ
+ជ
+ឈ
+ញ
+ដ
+ឋ
+ឌ
+ឍ
+ណ
+ត
+ថ
+ទ
+ធ
+ន
+ប
+ផ
+ព
+ភ
+ម
+យ
+រ
+ល
+វ
+ឝ
+ឞ
+ស
+ហ
+ឡ
+អ
+ឣ
+ឤ
+ឥ
+ឦ
+ឧ
+ឨ
+ឩ
+ឪ
+ឫ
+ឬ
+ឭ
+ឮ
+ឯ
+ឰ
+ឱ
+ឲ
+ឳ
+ា
+ិ
+ី
+ឹ
+ឺ
+ុ
+ូ
+ួ
+ើ
+ឿ
+ៀ
+េ
+ែ
+ៃ
+ោ
+ៅ
+ំ
+ះ
+ៈ
+៉
+៊
+់
+៌
+៍
+៎
+៏
+័
+៑
+្
+០
+១
+២
+៣
+៤
+៥
+៦
+៧
+៨
+៩
+។
+៕
+៖
+ៗ
+៘
+៙
+៚
+​
+‌
+‍

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a4b4f5a36fb5f5bd4a08164bf4d9d7bbcd3c2ce47f05f6a3e34bc8ef6620182
+size 6550612