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-large-nerd-latest"
|
| 30 |
+
tokenizer = "cardiffnlp/twitter-roberta-large-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)
|