hplisiecki
/

word2affect_polish

Text Classification

feature-extraction

Model card Files Files and versions Community

hplisiecki commited on Sep 13

Commit

a6b71d3

•

1 Parent(s): 90e0f97

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ model_directory = "word2affect_polish" # path to the cloned repository
 model = CustomModel.from_pretrained(model_directory)
 tokenizer = PreTrainedTokenizerFast.from_pretrained(model_directory)
-inputs = tokenizer("This is a test input.", return_tensors="pt")
 outputs = model(inputs['input_ids'], inputs['attention_mask'])
 # Print out the emotion ratings

 model = CustomModel.from_pretrained(model_directory)
 tokenizer = PreTrainedTokenizerFast.from_pretrained(model_directory)
+inputs = tokenizer("test", return_tensors="pt")
 outputs = model(inputs['input_ids'], inputs['attention_mask'])
 # Print out the emotion ratings