Update app.py

app.py

@@ -81,7 +81,7 @@ def generar_k_transpuesta_y_v(Q, palabra):
 def scaled_dot_product_attention_2D(Q, K_transpose, V):
     out = np.zeros((Q.shape[0], V.shape[1]))
     for i in range(Q.shape[0]):
-        scores = np.dot(Q[i, :], K_transpose.T)
+        scores = np.dot(Q.reshape(-1, 1), K_transpose)
         softmax_scores = np.exp(scores - np.max(scores)) / np.sum(np.exp(scores - np.max(scores)))
         out[i, :] = np.dot(softmax_scores, V)
     return out