Update README.md
Browse files
README.md
CHANGED
@@ -26,6 +26,9 @@ text = 'Banana+Nutella snack pack=someone is gonna see me crying in the break ro
|
|
26 |
definition = 'chocolate hazelnut spread manufactured by Ferrero'
|
27 |
target = 'Nutella'
|
28 |
|
|
|
|
|
|
|
29 |
text_input = f"{text} </s> {definition} </s> {target}"
|
30 |
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
31 |
pipe(text_input)
|
|
|
26 |
definition = 'chocolate hazelnut spread manufactured by Ferrero'
|
27 |
target = 'Nutella'
|
28 |
|
29 |
+
model = "cardiffnlp/twitter-roberta-base-nerd-latest"
|
30 |
+
tokenizer = "cardiffnlp/twitter-roberta-base-nerd-latest"
|
31 |
+
|
32 |
text_input = f"{text} </s> {definition} </s> {target}"
|
33 |
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
34 |
pipe(text_input)
|