Update app.py

app.py

@@ -112,7 +112,12 @@ def get_bert_embedding(review_text):
 #Get SpaBERT Embedding for geo-entity
 def get_spaBert_embedding(entity):
     entity_index = entity_index_dict.get(entity.lower(), None)
-    return spaBERT_embeddings[entity_index]
+    if entity_index is None:
+        st.write("Got Bert embedding for: ", entity)
+        return get_bert_embedding(text)
+    else:
+        st.write("Got SpaBert embedding for: ", entity)
+        return spaBERT_embeddings[entity_index]
 
 #Go through each review, identify all geo-entities, then extract their SpaBERT embedings
 def processSpatialEntities(review, nlp):
@@ -123,8 +128,8 @@ def processSpatialEntities(review, nlp):
     # Iterate over each entity span and process only geo entities
     for start, end, text, label in entity_spans:
         if label in ['FAC', 'ORG', 'LOC', 'GPE']:  # Filter to geo-entities
-            #spaBert_emb = get_spaBert_embedding(text)
-            #token_embeddings.append((text, spaBert_emb))
+            spaBert_emb = get_spaBert_embedding(text)
+            token_embeddings.append((text, spaBert_emb))
             st.write("Geo-Entity Found in review: ", text)
     return token_embeddings