Spaces:
Runtime error
Runtime error
Vignesh1997
commited on
Commit
•
5d70ea5
1
Parent(s):
90d6332
Update app.py
Browse files
app.py
CHANGED
@@ -160,8 +160,8 @@ def find_algnments(
|
|
160 |
source_target_mapping = nn.Softmax(dim=-1)(
|
161 |
torch.matmul(source_embedding, target_embedding)
|
162 |
)
|
163 |
-
target_source_mapping = nn.Softmax(dim=-
|
164 |
-
torch.matmul(target_embedding
|
165 |
)
|
166 |
align_matrix = (source_target_mapping > threshold) * (
|
167 |
target_source_mapping > threshold
|
|
|
160 |
source_target_mapping = nn.Softmax(dim=-1)(
|
161 |
torch.matmul(source_embedding, target_embedding)
|
162 |
)
|
163 |
+
target_source_mapping = nn.Softmax(dim=-2)(
|
164 |
+
torch.matmul(source_embedding,target_embedding)
|
165 |
)
|
166 |
align_matrix = (source_target_mapping > threshold) * (
|
167 |
target_source_mapping > threshold
|