cmunhozc
/

news-ranking-ft-bert

Text Classification

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

cmunhozc commited on Jan 7

Commit

c003d50

•

1 Parent(s): 82b5221

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -50,7 +50,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from transformers import Trainer
 ### 1. Load the model:
-model_name = "cmunhozc/news-ft-bert-cased"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 from transformers import Trainer
 ### 1. Load the model:
+model_name = "cmunhozc/news-ranking-ft-bert"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)