Update app.py

app.py

@@ -114,7 +114,7 @@ def get_spaBert_embedding(entity):
     entity_index = entity_index_dict.get(entity.lower(), None)
     if entity_index is None:
         st.write("Got Bert embedding for: ", entity)
-        return get_bert_embedding(text)
+        return get_bert_embedding(entity)
     else:
         st.write("Got SpaBert embedding for: ", entity)
         return spaBERT_embeddings[entity_index]