model update

README.md

@@ -5,7 +5,7 @@ metrics:
 - f1
 - accuracy
 model-index:
-- name: roberta-base-tweet-topic-multi-all
+- name: cardiffnlp/roberta-base-tweet-topic-multi-all
   results:
   - task:
       type: text-classification
@@ -52,8 +52,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 def sigmoid(x):
   return 1 / (1 + math.exp(-x))
   
-tokenizer = AutoTokenizer.from_pretrained(roberta-base-tweet-topic-multi-all)
-model = AutoModelForSequenceClassification.from_pretrained(roberta-base-tweet-topic-multi-all, problem_type="multi_label_classification")
+tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/roberta-base-tweet-topic-multi-all")
+model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/roberta-base-tweet-topic-multi-all", problem_type="multi_label_classification")
 model.eval()
 class_mapping = model.config.id2label