Update app.py

app.py

@@ -1,9 +1,11 @@
 import streamlit as st
 import spacy
 from transformers import BertTokenizer, BertModel
-
 from transformers.models.bert.modeling_bert import BertForMaskedLM
 
+from models.spabert.models.spatial_bert_model import SpatialBertConfig, SpatialBertForMaskedLM, SpatialBertModel
+from models.spabert.utils.common_utils import load_spatial_bert_pretrained_weights
+
 from PIL import Image
 
 ##LOAD MODEL SECTION##