Upload folder using huggingface_hub

app/model.py

@@ -6,10 +6,9 @@ from transformers import AlbertTokenizerFast, AlbertForSequenceClassification
 
 MODEL_REPO = "hadangvu/pkd-finbert-student"
 
-# Your label_mapping.json order
-LABELS = {0: "negative", 1: "neutral", 2: "positive"}
+# From label_mapping.json — order matches FinBERT teacher training
+LABELS = {0: "positive", 1: "negative", 2: "neutral"}
 
-# Runs ONCE when the server starts — not on every request
 print("Loading model from HF Hub...")
 tokenizer = AlbertTokenizerFast.from_pretrained(MODEL_REPO)
 model = AlbertForSequenceClassification.from_pretrained(MODEL_REPO)