KameronB
/

SITCC-Incident-Request-Classifier

natural-language

Model card Files Files and versions

KameronB commited on Apr 19, 2024

Commit

0daa01f

·

verified ·

1 Parent(s): b37695a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ import pandas as pd
 tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
 # Load RoBERTa pre-trained model
-model = RobertaForSequenceClassification.from_pretrained('roberta-base', num_labels=2)
 model = model.to(torch.device('cuda' if torch.cuda.is_available() else 'cpu'))

 tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
 # Load RoBERTa pre-trained model
+model = RobertaForSequenceClassification.from_pretrained('KameronB/SITCC-Incident-Request-Classifier', num_labels=2)
 model = model.to(torch.device('cuda' if torch.cuda.is_available() else 'cpu'))