anggars commited on
Commit
3abd738
·
verified ·
1 Parent(s): 5e534b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -19,7 +19,14 @@ print("Downloading model and extracting label encoders...")
19
  model_path = hf_hub_download(repo_id=REPO_ID, filename="model.pt")
20
  ckpt = torch.load(model_path, map_location=torch.device('cpu'), weights_only=False)
21
 
22
- label_encoders = ckpt['label_encoders']
 
 
 
 
 
 
 
23
  num_classes = {col: len(le.classes_) for col, le in label_encoders.items()}
24
 
25
  # ============================================================
 
19
  model_path = hf_hub_download(repo_id=REPO_ID, filename="model.pt")
20
  ckpt = torch.load(model_path, map_location=torch.device('cpu'), weights_only=False)
21
 
22
+ # Petain manual sesuai nama key pas lu nge-save model di PyTorch
23
+ label_encoders = {
24
+ 'mbti': ckpt['le_mbti'],
25
+ 'emotion': ckpt['le_emotion'],
26
+ 'vibe': ckpt['le_vibe'],
27
+ 'intensity': ckpt['le_intensity'],
28
+ 'tempo': ckpt['le_tempo']
29
+ }
30
  num_classes = {col: len(le.classes_) for col, le in label_encoders.items()}
31
 
32
  # ============================================================