Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|