Update README.md

README.md

@@ -62,25 +62,23 @@ First run
 
 ```bash
 git clone https://huggingface.co/hplisiecki/polemo_intensity
+```
 
+in order to clone the repository. Because of the custom model class, this model cannot be run with the usual huggingface Model setups.
 
-to
-
+Proceed as follows:
 
 ```python
+from polemo_intensity.model_script import Model
 from transformers import AutoTokenizer
-import torch
-
-# Load the tokenizer
-tokenizer = AutoTokenizer.from_pretrained("hplisiecki/polemo-intensity")
-
-# Load the model
-model = Model.from_pretrained("hplisiecki/polemo-intensity")
 
-# Test the model with a sample input
+model_directory = "polemo_intensity"  # Update this path
+model = Model.from_pretrained(model_directory)
+tokenizer = AutoTokenizer.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
 for emotion, rating in zip(['Happiness', 'Sadness', 'Anger', 'Disgust', 'Fear', 'Pride', 'Valence', 'Arousal'], outputs):
-    print(f"{emotion}: {rating.item()}")
+    print(f"{emotion}: {rating.item()}")
+```