Update README.md

README.md

@@ -29,7 +29,7 @@ model_name = "ESGBERT/EnvRoBERTa-environmental"
 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("Scope 1 emissions are reported here on a like-for-like basis against the 2013 baseline and exclude emissions from additional vehicles used during repairs."))