UngLong commited on
Commit
fa79eeb
·
verified ·
1 Parent(s): 7cd3bca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -26,4 +26,6 @@ This model is a fine-tuned version of CafeBERT on a toxic comment classification
26
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
27
 
28
  tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
29
- model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
 
 
 
26
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
27
 
28
  tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
29
+ model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
30
+
31
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/67ce9df626fb124ddb502190/2XHS-PzfNNEnGas6rm1iq.png)