selectorseb commited on
Commit
53befee
·
verified ·
1 Parent(s): 0c061a8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -20,7 +20,7 @@ Developed by [Selector AI](https://www.selector.ai/)
20
  from transformers import BertForSequenceClassification, BertTokenizer
21
 
22
  # Step 1: Load the model and tokenizer from Hugging Face
23
- model = BertForSequenceClassification.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
24
  tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
25
 
26
  import torch
 
20
  from transformers import BertForSequenceClassification, BertTokenizer
21
 
22
  # Step 1: Load the model and tokenizer from Hugging Face
23
+ model = BertForSequenceClassification.from_pretrained("SelectorAI/log-classifier-BERT-v1")
24
  tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
25
 
26
  import torch