SelectorAI
/

s2-log-classifier-BERT-v1

Text Classification

log-classification

Model card Files Files and versions

selectorseb commited on Dec 13, 2024

Commit

53befee

·

verified ·

1 Parent(s): 0c061a8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ Developed by [Selector AI](https://www.selector.ai/)
 from transformers import BertForSequenceClassification, BertTokenizer
 # Step 1: Load the model and tokenizer from Hugging Face
-model = BertForSequenceClassification.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
 tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
 import torch

 from transformers import BertForSequenceClassification, BertTokenizer
 # Step 1: Load the model and tokenizer from Hugging Face
+model = BertForSequenceClassification.from_pretrained("SelectorAI/log-classifier-BERT-v1")
 tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
 import torch