Update app.py

app.py

@@ -1,20 +1,20 @@
 import streamlit as st
 import spacy
 from transformers import BertTokenizer, BertModel
+
+from transformers.models.bert.modeling_bert import BertForMaskedLM
+
 from PIL import Image
 
-#LOAD MODEL SECTION#
+##LOAD MODEL SECTION##
 nlp = spacy.load("./models/en_core_web_sm")
 
+#BERT Section
 bert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
 bert_model = BertModel.from_pretrained("bert-base-uncased")
 bert_model.eval()
-st.write("Bert enabled - remove")
-
-
-
-
 
+#SpaBERT Section