Update README.md

docs: fixing Using Example stuff

README.md

@@ -94,16 +94,17 @@ import torch
 from transformers import RobertaTokenizer, RobertaForSequenceClassification
 
 
-model_name = 'metanovus/ruroberta-ecom-tech-best'
+model_name = 'ai-forever/ruRoberta-large'
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
 tokenizer = RobertaTokenizer.from_pretrained(model_name)
+class_descriptions = pd.read_csv('files/trends_description.csv')
 
 class BERTClass(torch.nn.Module):
     def __init__(self):
         super(BERTClass, self).__init__()
         self.bert_model = RobertaForSequenceClassification.from_pretrained(
-            model_name,
+            'metanovus/ruroberta-ecom-tech-best',
             return_dict=True,
             problem_type='multi_label_classification',
             num_labels=50