Update app.py

app.py

@@ -87,9 +87,16 @@ selected_review = example_reviews[user_input]
 # Process the text when the button is clicked
 if st.button("Highlight Geo-Entities"):
     if selected_review.strip():
+        bert_embedding = get_bert_embedding(selected_review)
+        # Debug: Print the shape of the embeddings
+        st.write("Embedding Shape:", bert_embedding.shape)
+        
+        # Debug: Print the embeddings themselves (optional)
+        st.write("Embeddings:", bert_embedding)
+        
         # Process the text using spaCy
         doc = nlp(selected_review)
-
+        
         # Highlight geo-entities with different colors
         highlighted_text = selected_review
         for ent in reversed(doc.ents):