tasinhoque
/

text-classification-goemotions

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

tasinhoque commited on Feb 27, 2023

Commit

ba8ad0c

•

1 Parent(s): 4b9e5e3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ model = AutoModelForSequenceClassification.from_pretrained(
     num_labels=n_emotion,
     problem_type="multi_label_classification",
     revision=commit
-).to(device)
 ```
 ## Intended uses & limitations

     num_labels=n_emotion,
     problem_type="multi_label_classification",
     revision=commit
+)
 ```
 ## Intended uses & limitations