Commit
•
d25036a
1
Parent(s):
38e867d
Update README.md
Browse files
README.md
CHANGED
|
@@ -40,8 +40,10 @@ from word2affect_spanish.model_script import CustomModel # importing the custom
|
|
| 40 |
from transformers import PreTrainedTokenizerFast
|
| 41 |
|
| 42 |
model_directory = "word2affect_spanish" # path to the cloned repository
|
|
|
|
| 43 |
model = CustomModel.from_pretrained(model_directory)
|
| 44 |
tokenizer = PreTrainedTokenizerFast.from_pretrained(model_directory)
|
|
|
|
| 45 |
inputs = tokenizer("This is a test input.", return_tensors="pt")
|
| 46 |
outputs = model(inputs['input_ids'], inputs['attention_mask'])
|
| 47 |
|
|
|
|
| 40 |
from transformers import PreTrainedTokenizerFast
|
| 41 |
|
| 42 |
model_directory = "word2affect_spanish" # path to the cloned repository
|
| 43 |
+
|
| 44 |
model = CustomModel.from_pretrained(model_directory)
|
| 45 |
tokenizer = PreTrainedTokenizerFast.from_pretrained(model_directory)
|
| 46 |
+
|
| 47 |
inputs = tokenizer("This is a test input.", return_tensors="pt")
|
| 48 |
outputs = model(inputs['input_ids'], inputs['attention_mask'])
|
| 49 |
|