Update encoder model card

README.md

@@ -28,6 +28,7 @@ weights_path = weights_path_for("encoder")
 download_weights(
     "encoder",
     weights_path,
+    repo_id="ryeyoo/sentimentizer-encoder",
     dict_path=DriverConfig.files.dictionary_file_path,
 )
 
@@ -38,10 +39,10 @@ from sentimentizer.tokenizer import get_trained_tokenizer
 model = get_trained_model(device="cpu")
 tokenizer = get_trained_tokenizer()
 
-import numpy as np
-token_ids = tokenizer.tokenize_text("amazing food great service")
-score = model.predict(token_ids)
-print(f'Sentiment score: {score.item():.4f}')  # >0.5 = positive, <0.5 = negative
+probs = model.predict_text('amazing food great service')
+for label, prob in sorted(probs.items(), key=lambda x: -x[1]):
+    print(f'{label}: {prob:.4f}')
+# e.g. positive: 0.8300, neutral: 0.1200, negative: 0.0500
 ```
 
 ## Files