Update models/preprocess_stage/bert_model.py

models/preprocess_stage/bert_model.py

@@ -2,14 +2,14 @@ import numpy as np
 import torch
 from transformers import BertModel, BertTokenizer
 
-weights = 'DeepPavlov/rubert-base-cased'
-tokenizer_bert = BertTokenizer.from_pretrained(weights)
-bert_for_logreg = BertModel.from_pretrained(weights)
+CHECKPOINT = 'DeepPavlov/rubert-base-cased'
+tokenizer = BertTokenizer.from_pretrained(CHECKPOINT)
+model = BertModel.from_pretrained(CHECKPOINT)
 
 
 def preprocess_bert(text, MAX_LEN):
 
-    tokenized_text = tokenizer_bert.encode(
+    tokenized_text = tokenizer.encode(
         text=text,
         add_special_tokens=True,
         truncation=True,