Update README.md

README.md

@@ -26,10 +26,8 @@ To use the model for making predictions, simply load the model and tokenizer, th
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
-# Set up the device (GPU if available, otherwise CPU)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# Label mapping dictionary
 label_mapping = {
     0: "No need for visual processing",
     1: "Requires visual processing",