FaiziRBLX commited on
Commit
1aa267f
·
verified ·
1 Parent(s): 511d3ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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!