faimlab
/

fabert_formality_classifier

Text Classification

Model card Files Files and versions Community

faimsut commited on Jan 29

Commit

c2c53f0

·

verified ·

1 Parent(s): 8235ddb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -35,8 +35,8 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 # Load the pre-trained tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("FAIMs/fabert_formality_classifier")
-model = AutoModelForSequenceClassification.from_pretrained("FAIMs/fabert_formality_classifier")
 # Ensure the model runs on GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 # Load the pre-trained tokenizer and model
+tokenizer = AutoTokenizer.from_pretrained("faimlab/fabert_formality_classifier")
+model = AutoModelForSequenceClassification.from_pretrained("faimlab/fabert_formality_classifier")
 # Ensure the model runs on GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")