Update README.md

README.md

@@ -34,7 +34,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import torch.nn.functional as F
 
-model_name = "your-username/your-model-repo"
+model_name = "RiyaSirohi/bert-base-lora-6class"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 model.eval()
@@ -51,5 +51,5 @@ print(label, probs)
 
 ## Limitations
 
-The model may misclassify subtle or sarcastic inputs.
-Fine-tuned with a small number of epochs — more training may improve performance.
+- The model may misclassify subtle or sarcastic inputs.
+- Fine-tuned with a small number of epochs — more training may improve performance.