hplisiecki
/

word2affect_spanish

Text Classification

feature-extraction

Model card Files Files and versions Community

hplisiecki commited on Jun 18

Commit

38e867d

•

1 Parent(s): bae0f76

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -40,8 +40,8 @@ from word2affect_spanish.model_script import CustomModel # importing the custom
 from transformers import PreTrainedTokenizerFast
 model_directory = "word2affect_spanish" # path to the cloned repository
-model = CustomModel.from_pretrained('word2affect_spanish')
-tokenizer = PreTrainedTokenizerFast.from_pretrained('word2affect_spanish')
 inputs = tokenizer("This is a test input.", return_tensors="pt")
 outputs = model(inputs['input_ids'], inputs['attention_mask'])

 from transformers import PreTrainedTokenizerFast
 model_directory = "word2affect_spanish" # 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'])