hplisiecki commited on
Commit
88e20ed
·
verified ·
1 Parent(s): 44dbbd9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -50,7 +50,7 @@ model_directory = "word2affect_english" # path to the cloned repository
50
  model = CustomModel.from_pretrained(model_directory)
51
  tokenizer = AutoTokenizer.from_pretrained(model_directory)
52
 
53
- inputs = tokenizer("This is a test input.", return_tensors="pt")
54
  outputs = model(inputs['input_ids'], inputs['attention_mask'])
55
 
56
  # Print out the emotion ratings
 
50
  model = CustomModel.from_pretrained(model_directory)
51
  tokenizer = AutoTokenizer.from_pretrained(model_directory)
52
 
53
+ inputs = tokenizer("test", return_tensors="pt")
54
  outputs = model(inputs['input_ids'], inputs['attention_mask'])
55
 
56
  # Print out the emotion ratings