selectorseb commited on
Commit
9a39249
·
verified ·
1 Parent(s): d7924f4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -20,8 +20,8 @@ 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("SelectorAI/log-classifier-BERT-v1")
24
- tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
25
 
26
  import torch
27
 
 
20
  from transformers import BertForSequenceClassification, BertTokenizer
21
 
22
  # Step 1: Load the model and tokenizer from Hugging Face
23
+ model = BertForSequenceClassification.from_pretrained("SelectorAI/s2-log-classifier-BERT-v1")
24
+ tokenizer = BertTokenizer.from_pretrained("SelectorAI/s2-log-classifier-BERT-v1")
25
 
26
  import torch
27