Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ language:
|
|
6 |
- en
|
7 |
pipeline_tag: text-classification
|
8 |
---
|
9 |
-
# cardiffnlp/twitter-roberta-large-latest
|
10 |
|
11 |
This is a RoBERTa-large model trained on 154M tweets until the end of December 2022 and finetuned for topic classification (multilabel classification) on the _TweetTopic_ dataset of [SuperTweetEval](https://huggingface.co/datasets/cardiffnlp/super_tweeteval).
|
12 |
The original Twitter-based RoBERTa model can be found [here](https://huggingface.co/cardiffnlp/twitter-roberta-large-2022-154m).
|
@@ -42,7 +42,7 @@ The original Twitter-based RoBERTa model can be found [here](https://huggingface
|
|
42 |
from transformers import pipeline
|
43 |
text = "So @AB is just the latest victim of the madden curse. If you’re on the cover of that game your career will take a turn for the worse"
|
44 |
|
45 |
-
pipe = pipeline('text-classification', model="cardiffnlp/twitter-roberta-large-latest
|
46 |
predictions = pipe(text)[0]
|
47 |
predictions = [x for x in predictions if x['score'] > 0.5]
|
48 |
predictions
|
|
|
6 |
- en
|
7 |
pipeline_tag: text-classification
|
8 |
---
|
9 |
+
# cardiffnlp/twitter-roberta-large-topic-latest
|
10 |
|
11 |
This is a RoBERTa-large model trained on 154M tweets until the end of December 2022 and finetuned for topic classification (multilabel classification) on the _TweetTopic_ dataset of [SuperTweetEval](https://huggingface.co/datasets/cardiffnlp/super_tweeteval).
|
12 |
The original Twitter-based RoBERTa model can be found [here](https://huggingface.co/cardiffnlp/twitter-roberta-large-2022-154m).
|
|
|
42 |
from transformers import pipeline
|
43 |
text = "So @AB is just the latest victim of the madden curse. If you’re on the cover of that game your career will take a turn for the worse"
|
44 |
|
45 |
+
pipe = pipeline('text-classification', model="cardiffnlp/twitter-roberta-large-topic-latest", return_all_scores=True)
|
46 |
predictions = pipe(text)[0]
|
47 |
predictions = [x for x in predictions if x['score'] > 0.5]
|
48 |
predictions
|