Update app.py

app.py

@@ -1,9 +1,39 @@
 import streamlit as st
 import spacy
-from transformers import pipeline
+from transformers import BertTokenizer, BertModel
 from PIL import Image
 
+#LOAD MODEL SECTION#
 nlp = spacy.load("./models/en_core_web_sm")
+
+bert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
+bert_model = BertModel.from_pretrained("bert-base-uncased")
+bert_model.eval()
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 
 st.title("SpaGAN Demo")
 st.write("Enter a text, and the system will highlight the geo-entities within it.")