update doc

README.md

@@ -14,18 +14,23 @@ This model is fine-tuned from `microsoft/Multilingual-MiniLM-L12-H384` to classi
 
 ## Usage
 
+This model is fine-tuned from `microsoft/Multilingual-MiniLM-L12-H384` to classify email subjects as SPAM or NOSPAM.
+
+### Loading the model
+To load this model, ensure you specify `from_safetensors=True`:
+
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 model_name = "Goodmotion/spam-mail-classifier"
-
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(
+    model_name, from_safetensors=True
+)
 
 text = "Félicitations ! Vous avez gagné un iPhone."
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model(**inputs)
-
 print(outputs.logits)
 ```
 
@@ -38,7 +43,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model_name = "Goodmotion/spam-mail-classifier"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name,from_safetensors=True)
 
 texts = [
 'Join us for a webinar on AI innovations',

config.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e1f5d888b080781be08c900cb8478a9024c698e80161535b4112b298674d6539
-size 769
+oid sha256:9dad0e77cb2fdbd8f66acdc9e26e06651be2490dfcdf6e428bb09f1aaf5df7cd
+size 796