Update README.md

README.md

@@ -29,7 +29,7 @@ model_name = "ESGBERT/SocRoBERTa-social"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, max_len=512)
  
-pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0)
+pipe = pipeline("text-classification", model=model, tokenizer=tokenizer) # set device=0 to use GPU
  
 # See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
 print(pipe("We follow rigorous supplier checks to prevent slavery and ensure workers' rights."))