Update README.md
Browse files
README.md
CHANGED
@@ -42,8 +42,8 @@ You can use the model like this:
|
|
42 |
>>> 'Search Screening Event'
|
43 |
>>> ]
|
44 |
>>> aspect = 'intent'
|
45 |
-
>>>
|
46 |
-
>>> text = f'{aspect} {
|
47 |
|
48 |
>>> query = [[text, lb] for lb in labels]
|
49 |
>>> logits = model.predict(query, apply_softmax=True)
|
|
|
42 |
>>> 'Search Screening Event'
|
43 |
>>> ]
|
44 |
>>> aspect = 'intent'
|
45 |
+
>>> aspect_sep_token = model.tokenizer.additional_special_tokens[0]
|
46 |
+
>>> text = f'{aspect} {aspect_sep_token} {text}'
|
47 |
|
48 |
>>> query = [[text, lb] for lb in labels]
|
49 |
>>> logits = model.predict(query, apply_softmax=True)
|