Update app.py

app.py

@@ -136,8 +136,9 @@ def processSpatialEntities(review, nlp):
             spaBert_emb = get_spaBert_embedding(text)
             token_embeddings.append(spaBert_emb)
             st.write("Geo-Entity Found in review: ", text)
-            
-    processed_embedding = torch.cat(token_embeddings, dim=0)
+
+    token_embeddings = torch.stack(token_embeddings, dim=0)
+    processed_embedding = token_embeddings.mean(dim=0, keepdim=True)
     st.write("processed embedding shape: ", processed_embedding.shape)
     return processed_embedding