JasonTPhillipsJr commited on
Commit
b6c17e6
1 Parent(s): bec805a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def processSpatialEntities(review, nlp):
139
 
140
  token_embeddings = torch.stack(token_embeddings, dim=0)
141
  processed_embedding = token_embeddings.mean(dim=0) # Shape: (768)
142
- processed_embedding = processed_embedding.unsqueeze(0) # Shape: (1, 768)
143
  return processed_embedding
144
 
145
 
 
139
 
140
  token_embeddings = torch.stack(token_embeddings, dim=0)
141
  processed_embedding = token_embeddings.mean(dim=0) # Shape: (768)
142
+ #processed_embedding = processed_embedding.unsqueeze(0) # Shape: (1, 768)
143
  return processed_embedding
144
 
145