Update app.py

app.py

@@ -9,14 +9,15 @@ st.markdown("### Articles classificator.")
 @st.cache
 def get_bert_and_tokenizer():
     model_name = 'bert-base-uncased'
-    return AutoModel.from_pretrained(model_name), AutoTokenizer.from_pretrained(model_name)
+    # return AutoModel.from_pretrained(model_name), AutoTokenizer.from_pretrained(model_name)
+    return AutoTokenizer.from_pretrained(model_name)
 
-bert, tokenizer = get_bert_and_tokenizer()
+tokenizer = get_bert_and_tokenizer()
 
 class devops_model(nn.Module):
     def __init__(self):
         super(devops_model, self).__init__()
-        self.bert = bert
+        self.bert = None
         self.fc = nn.Sequential(
             nn.Linear(768, 768),
             nn.ReLU(),