r0ymanesco
commited on
Commit
•
fe77b27
1
Parent(s):
15bc23d
Update README.md
Browse filesfix model path string
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("
|
47 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
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")
|