Update app.py

app.py

@@ -79,7 +79,7 @@ def compute_splade_vector(text: str) -> models.SparseVector:
     with torch.no_grad():
         output = splade_model(**tokens)
     
-    logits, attention_mask = output.logits, tokens.attention_mask
+    logits, attention_mask = output.logits, tokens['attention_mask']
     relu_log = torch.log(1 + torch.relu(logits))
     weighted_log = relu_log * attention_mask.unsqueeze(-1)
     max_val, _ = torch.max(weighted_log, dim=1)