tobischimanski commited on
Commit
03ce76a
1 Parent(s): e33af34

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ model_name = "ESGBERT/SocRoBERTa-social"
29
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
30
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, max_len=512)
31
 
32
- pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0)
33
 
34
  # See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
35
  print(pipe("We follow rigorous supplier checks to prevent slavery and ensure workers' rights."))
 
29
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
30
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, max_len=512)
31
 
32
+ pipe = pipeline("text-classification", model=model, tokenizer=tokenizer) # set device=0 to use GPU
33
 
34
  # See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
35
  print(pipe("We follow rigorous supplier checks to prevent slavery and ensure workers' rights."))