JasonTPhillipsJr commited on
Commit
cb96047
·
verified ·
1 Parent(s): 843e38a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ def processSpatialEntities(review, nlp):
134
  for start, end, text, label in entity_spans:
135
  if label in ['FAC', 'ORG', 'LOC', 'GPE']: # Filter to geo-entities
136
  spaBert_emb = get_spaBert_embedding(text)
137
- token_embeddings.append((text, spaBert_emb))
138
  st.write("Geo-Entity Found in review: ", text)
139
 
140
  processed_embedding = torch.cat(token_embeddings, dim=0)
 
134
  for start, end, text, label in entity_spans:
135
  if label in ['FAC', 'ORG', 'LOC', 'GPE']: # Filter to geo-entities
136
  spaBert_emb = get_spaBert_embedding(text)
137
+ token_embeddings.append(spaBert_emb)
138
  st.write("Geo-Entity Found in review: ", text)
139
 
140
  processed_embedding = torch.cat(token_embeddings, dim=0)