Update README.md

README.md

@@ -33,7 +33,7 @@ You can use the model and tokenizer as follows:
 First run the bash code below to clone the repository (this will take some time). Because of the custom model class, this model cannot be run with the usual huggingface Model setups.
 
 ```bash
-git clone https://huggingface.co/hplisiecki/polemo_intensity
+git clone https://huggingface.co/hplisiecki/word2affect_polish
 ```
 
 Proceed as follows:
@@ -43,8 +43,10 @@ from word2affect_polish.model_script import CustomModel # importing the custom m
 from transformers import PreTrainedTokenizerFast
 
 model_directory = "word2affect_polish" # path to the cloned repository
-model = CustomModel.from_pretrained('word2affect_polish')
-tokenizer = PreTrainedTokenizerFast.from_pretrained('word2affect_polish')
+
+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'])