pass to safetensor

README.md

@@ -16,16 +16,13 @@ This model is fine-tuned from `microsoft/Multilingual-MiniLM-L12-H384` to classi
 
 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, from_safetensors=True
+    model_name
 )
 
 text = "Félicitations ! Vous avez gagné un iPhone."
@@ -43,7 +40,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model_name = "Goodmotion/spam-mail-classifier"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name,from_safetensors=True)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 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:9dad0e77cb2fdbd8f66acdc9e26e06651be2490dfcdf6e428bb09f1aaf5df7cd
-size 796
+oid sha256:e1f5d888b080781be08c900cb8478a9024c698e80161535b4112b298674d6539
+size 769