Kendamarron
/

fineweb-edu-classifier-ja

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Kendamarron commited on 13 days ago

Commit

0793e36

•

1 Parent(s): 00f677e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ pipeline_tag: text-classification
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("Kendamarron/fineweb-edu-classifier-ja")
-model = AutoModelForSequenceClassification.from_pretrained("Kendamarron/fineweb-edu-classifier-ja", num_labels=6, classifier_dropout=0.0, hidden_dropout_prob=0.0)
 def predict(text):
     inputs = tokenizer(text, return_tensors="pt")

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("Kendamarron/fineweb-edu-classifier-ja")
+model = AutoModelForSequenceClassification.from_pretrained("Kendamarron/fineweb-edu-classifier-ja")
 def predict(text):
     inputs = tokenizer(text, return_tensors="pt")