Update app.py

app.py

@@ -191,8 +191,7 @@ if st.button("Highlight Geo-Entities"):
         spaBert_embedding = processSpatialEntities(selected_review,nlp)
         st.write("Geo-Entities embedding shape: ", spaBert_embedding.shape)
 
-        #combine the embeddings (NOTE: come back and update after testing)
-        combined_embedding = torch.cat((bert_embedding,spaBERT_embeddings[0]),dim=-1)
+        combined_embedding = torch.cat((bert_embedding,spaBert_embedding),dim=-1)
         st.write("Concatenated Embedding Shape:", combined_embedding.shape)
         st.write("Concatenated Embedding:", combined_embedding)