nie3e
/

go-emotions-polish-gpt2-small-v0.0.1

Text Classification

Generated from Trainer

multi-class-classification

multi-label-classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nie3e commited on 26 days ago

Commit

0484da9

·

1 Parent(s): 2f80df4

Adds model device

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -78,7 +78,7 @@ checkpoint = "nie3e/go-emotions-polish-gpt2-small-v0.0.1"
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 model = AutoModelForSequenceClassification.from_pretrained(
     checkpoint, problem_type="multi_label_classification"
-)
 text = "To jest model wykrywający super emocje w tekście! :D"

 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 model = AutoModelForSequenceClassification.from_pretrained(
     checkpoint, problem_type="multi_label_classification"
+).to(device)
 text = "To jest model wykrywający super emocje w tekście! :D"