Update app.py

app.py

@@ -4,22 +4,28 @@ from transformers import AutoTokenizer
 
 # IMPORT ARSITEKTUR DARI best.py
 # (Sesuaikan nama class-nya dengan yang ada di best.py, misalnya 'ModelConfig' dan 'MyTransformer')
-from best import ModelConfig, IndonesianLLM # <--- Ganti MyTransformer dengan nama class modelmu
+# =======================================================
+# 1. BANGUN KERANGKA MODELNYA DULU (Jangan sampai terhapus)
+# =======================================================
+config = ModelConfig() 
+model = IndonesianLLM(config) # <--- PENTING: Ganti 'NamaClassModelKamu' sesuai dengan nama class yang ada di best.py!
 
-# 1. Load Tokenizer (Sesuai dengan yang ada di best.py)
-# 1. Load file bobotnya dulu ke dalam variabel
+# =======================================================
+# 2. LOAD BOBOT MODEL DAN PERBAIKI NAMANYA
+# =======================================================
 state_dict = torch.load("model.pt", map_location=torch.device('cpu'), weights_only=False)
-# 2. Buat kamus baru untuk menyimpan nama kunci yang sudah diperbaiki
+
 new_state_dict = {}
 for key, value in state_dict.items():
-    # Jika nama kuncinya berawalan 'model.', kita potong 6 huruf pertamanya
     if key.startswith('model.'):
         new_key = key[6:] 
         new_state_dict[new_key] = value
     else:
         new_state_dict[key] = value
 
-# 3. Masukkan bobot yang namanya sudah diperbaiki ke dalam kerangka model
+# =======================================================
+# 3. MASUKKAN BOBOT KE DALAM KERANGKA MODEL
+# =======================================================
 model.load_state_dict(new_state_dict)
 model.eval()