Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,10 @@ from best import ModelConfig, IndonesianLLM
|
|
| 7 |
# (Sesuaikan nama class-nya dengan yang ada di best.py, misalnya 'ModelConfig' dan 'MyTransformer')
|
| 8 |
# =======================================================
|
| 9 |
# 1. BANGUN KERANGKA MODELNYA DULU (Jangan sampai terhapus)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
# =======================================================
|
| 11 |
config = ModelConfig()
|
| 12 |
model = IndonesianLLM(config) # <--- PENTING: Ganti 'NamaClassModelKamu' sesuai dengan nama class yang ada di best.py!
|
|
|
|
| 7 |
# (Sesuaikan nama class-nya dengan yang ada di best.py, misalnya 'ModelConfig' dan 'MyTransformer')
|
| 8 |
# =======================================================
|
| 9 |
# 1. BANGUN KERANGKA MODELNYA DULU (Jangan sampai terhapus)
|
| 10 |
+
|
| 11 |
+
# Load tokenizer
|
| 12 |
+
tokenizer = AutoTokenizer.from_pretrained("indolem/indobert-base-uncased")
|
| 13 |
+
tokenizer.add_special_tokens({"additional_special_tokens": ["<cot>", "</cot>"]})
|
| 14 |
# =======================================================
|
| 15 |
config = ModelConfig()
|
| 16 |
model = IndonesianLLM(config) # <--- PENTING: Ganti 'NamaClassModelKamu' sesuai dengan nama class yang ada di best.py!
|