r0ymanesco commited on
Commit
fe77b27
1 Parent(s): 15bc23d

Update README.md

Browse files

fix model path string

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,8 +43,8 @@ You can then determine the model to call as follows
43
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
44
 
45
  id2label = {0: 'gpt-3.5', 1: 'gpt-4'}
46
- tokenizer = AutoTokenizer.from_pretrained("r0ymanesco/notdiamond-0001")
47
- model = AutoModelForSequenceClassification.from_pretrained("r0ymanesco/notdiamond-0001")
48
  max_length = self._get_max_length(model)
49
 
50
  inputs = tokenizer(formatted_prompt, truncation=True, max_length=max_length, return_tensors="pt")
 
43
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
44
 
45
  id2label = {0: 'gpt-3.5', 1: 'gpt-4'}
46
+ tokenizer = AutoTokenizer.from_pretrained("notdiamond/notdiamond-0001")
47
+ model = AutoModelForSequenceClassification.from_pretrained("notdiamond/notdiamond-0001")
48
  max_length = self._get_max_length(model)
49
 
50
  inputs = tokenizer(formatted_prompt, truncation=True, max_length=max_length, return_tensors="pt")