tobischimanski
commited on
Commit
•
03ce76a
1
Parent(s):
e33af34
Update README.md
Browse files
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
|
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."))
|