Spaces:
Sleeping
Sleeping
JasonTPhillipsJr
commited on
Commit
•
b6c17e6
1
Parent(s):
bec805a
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 |
|