tum-nlp
/

Deberta_Human_Value_Detector

Text Classification

deberta_arg_classifier

Model card Files Files and versions Community

danschr commited on Nov 16, 2023

Commit

dc635b2

•

1 Parent(s): ce1d6e9

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -15,6 +15,7 @@ To use the model please follow the steps below...
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 tokenizer =  AutoTokenizer.from_pretrained("tum-nlp/Deberta_Human_Value_Detector")
 trained_model = AutoModelForSequenceClassification.from_pretrained("tum-nlp/Deberta_Human_Value_Detector", trust_remote_code=True)

 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
 tokenizer =  AutoTokenizer.from_pretrained("tum-nlp/Deberta_Human_Value_Detector")
 trained_model = AutoModelForSequenceClassification.from_pretrained("tum-nlp/Deberta_Human_Value_Detector", trust_remote_code=True)