textgain
/

FactRank_e5_small

Text Classification

Model card Files Files and versions

MKlioc commited on Mar 26, 2025

Commit

fb10856

·

verified ·

1 Parent(s): 38033b6

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -35,7 +35,6 @@ login(token=hf_token)
 config = AutoConfig.from_pretrained("textgain/FactRank_e5_small")
 tokenizer = AutoTokenizer.from_pretrained("textgain/FactRank_e5_small")
 model = AutoModelForSequenceClassification.from_pretrained("textgain/FactRank_e5_small", config=config)
-model.eval()
 pipe = pipeline(model=model, tokenizer=tokenizer, task="text-classification")

 config = AutoConfig.from_pretrained("textgain/FactRank_e5_small")
 tokenizer = AutoTokenizer.from_pretrained("textgain/FactRank_e5_small")
 model = AutoModelForSequenceClassification.from_pretrained("textgain/FactRank_e5_small", config=config)
 pipe = pipeline(model=model, tokenizer=tokenizer, task="text-classification")