Update app.py

app.py

@@ -229,6 +229,9 @@ if st.button("Highlight Geo-Entities"):
         combined_embedding = torch.cat((bert_embedding,spaBert_embedding),dim=-1)
         st.write("Concatenated Embedding Shape:", combined_embedding.shape)
         st.write("Concatenated Embedding:", combined_embedding)
+
+        prediction = discriminator(combined_embedding)
+        st.write("Prediction:", prediction)
         
         # Process the text using spaCy
         doc = nlp(selected_review)