Update README.md

README.md

@@ -2,17 +2,19 @@
 language: en
 license: other
 tags:
-  - sentiment-analysis
-  - fine-tuned
-  - sentiment-classification
-  - transformers
+- sentiment-analysis
+- fine-tuned
+- sentiment-classification
+- transformers
 model_name: Fine-Tuned Sentiment Model
 model_type: Roberta
 datasets:
-  - custom-dataset
+- custom-dataset
+- rohittamidapati11/training_data
+- rohittamidapati11/validation_data
 metrics:
-  - micro precision and recall
-  - macro precision and recall
+- micro precision and recall
+- macro precision and recall
 ---
 
 # Fine-Tuned Sentiment Model
@@ -48,4 +50,4 @@ model = AutoModelForSequenceClassification.from_pretrained("your_username/fine_t
 inputs = tokenizer("The food was a bit bland, but the portion sizes were generous. I was looking forward to trying it, but it didn't quite live up to my expectations.", return_tensors='pt')
 outputs = model(**inputs)
 predicted_class = torch.argmax(outputs.logits, dim = 1).item()
-print("Predicted Sentiment:", predicted_class)
+print("Predicted Sentiment:", predicted_class)