Update backend/app/train.py

backend/app/train.py

@@ -47,9 +47,8 @@ def tokenize_function(example, tokenizer):
     tokens["label"] = example["label"]  # ✅ Keep label after tokenization
     return tokens
 
-
 # -------- Main Training Function --------
-deTruest ain():
+def train():
     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
     model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, num_labels=3)
 
@@ -83,6 +82,6 @@ deTruest ain():
     tokenizer.save_pretrained(SAVE_PATH)
     print("✅ Fine-tuned model saved!")
 
+# -------- Entry Point --------
 if __name__ == "__main__":
-    train()
-
+    train()