Update README.md

README.md

@@ -26,7 +26,7 @@ This model classifies emotions in English text. It outputs the probability of fi
 
 ### Model Sources
 
-- **Repository**: [https://huggingface.co/roblesadrian/llama-3b-emotion-classifier]
+- **Repository**: [https://huggingface.co/gsi-upm/llama-3b-emotion-classifier]
   
 ---
 
@@ -57,8 +57,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 # Load the model and tokenizer
-model = AutoModelForSequenceClassification.from_pretrained("your-username/llama-3.2b-emotion-classifier")
-tokenizer = AutoTokenizer.from_pretrained("your-username/llama-3.2b-emotion-classifier")
+model = AutoModelForSequenceClassification.from_pretrained("gsi-upm/llama-3b-emotion-classifier", num_labels=5)
+tokenizer = AutoTokenizer.from_pretrained("gsi-upm/llama-3b-emotion-classifier")
 
 # Input text
 text = "I just got a promotion and I feel amazing!"
@@ -101,13 +101,6 @@ The model was trained on the **SemEval 2025** dataset, which consists of English
 
 - **Labels**: anger, fear, joy, sadness, surprise
 
-### Training Procedure
-
-- **Training regime**: Mixed precision (FP16) training using the `transformers` library.
-- **Optimizer**: AdamW
-- **Epochs**: 4
-- **Batch size**: 1
-
 ---
 
 ## Evaluation
@@ -130,17 +123,17 @@ If you use this model, please cite the following:
 
 ```
 @misc{llama-emotion-classifier,
-  author = {Adrián Maldonado Robles},
+  author = {GSI-UPM},
   title = {LLaMA-3.2B Fine-tuned for Emotion Classification},
   year = {2025},
   publisher = {Hugging Face},
-  url = {https://huggingface.co/roblesadrian/llama-3.2b-emotion-classifier}
+  url = {https://huggingface.co/gsi-upm/llama-3b-emotion-classifier}
 }
 ```
 
 **APA:**
 
-Adrián Maldonado Robles. (2025). *LLaMA-3.2B Fine-tuned for Emotion Classification*. Hugging Face. Available at: https://huggingface.co/your-username/llama-3.2b-emotion-classifier
+Adrián Maldonado Robles. (2025). *LLaMA-3.2B Fine-tuned for Emotion Classification*. Hugging Face. Available at: https://huggingface.co/gsi-upm/llama-3b-emotion-classifier
 
 ---