Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
# ============================================================
|