Update app.py

app.py

@@ -125,8 +125,9 @@ 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
 
 # Function to read reviews from a text file
@@ -184,7 +185,9 @@ if st.button("Highlight Geo-Entities"):
         st.write("Embedding Shape:", bert_embedding.shape)
         
         # Debug: Print the embeddings themselves (optional)
-        st.write("Embeddings:", bert_embedding)
+        #st.write("Embeddings:", bert_embedding)
+
+        spaBert_embedding = processSpatialEntities(selected_review,nlp)
 
         #combine the embeddings (NOTE: come back and update after testing)
         combined_embedding = torch.cat((bert_embedding,spaBERT_embeddings[0]),dim=-1)