AventIQ-AI
/

distilbert-spam-detection

Model card Files Files and versions Community

ayushsinha commited on Feb 10

Commit

ea847ee

·

verified ·

1 Parent(s): d75fac7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ pip install transformers torch
 ### Loading the Model
 ```python
-from transformers import DistilBertTokenizer, DistilBertForSequenceClassification, Trainer, TrainingArguments
 import torch
 model_name = "AventIQ-AI/distilbert-spam-detection"

 ### Loading the Model
 ```python
+from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
 import torch
 model_name = "AventIQ-AI/distilbert-spam-detection"