Update app.py

app.py

@@ -160,8 +160,8 @@ def find_algnments(
     source_target_mapping = nn.Softmax(dim=-1)(
         torch.matmul(source_embedding, target_embedding)
     )
-    target_source_mapping = nn.Softmax(dim=-1)(
-        torch.matmul(target_embedding, source_embedding)
+    target_source_mapping = nn.Softmax(dim=-2)(
+        torch.matmul(source_embedding,target_embedding)
     )
     align_matrix = (source_target_mapping > threshold) * (
         target_source_mapping > threshold