Hugging Face's logo

faimlab
/
fabert_formality_classifier

Text Classification
Safetensors
Persian
bert
code
Model card Files
xet
Community
faimsut commited on
Commit
c2c53f0
·
verified ·
1 Parent(s): 8235ddb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,8 +35,8 @@ import torch
35
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
36
 
37
  # Load the pre-trained tokenizer and model
38
- tokenizer = AutoTokenizer.from_pretrained("FAIMs/fabert_formality_classifier")
39
- model = AutoModelForSequenceClassification.from_pretrained("FAIMs/fabert_formality_classifier")
40
 
41
  # Ensure the model runs on GPU if available
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
35
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
36
 
37
  # Load the pre-trained tokenizer and model
38
+ tokenizer = AutoTokenizer.from_pretrained("faimlab/fabert_formality_classifier")
39
+ model = AutoModelForSequenceClassification.from_pretrained("faimlab/fabert_formality_classifier")
40
 
41
  # Ensure the model runs on GPU if available
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")