Update app.py

app.py

@@ -115,6 +115,7 @@ def infer(model, replay_file,
     # Set index name
     preds.index.name = "Frame"
     remove_ties_mask = is_ot if not ignore_ties else torch.ones(len(preds), dtype=torch.bool)
+    remove_ties_mask = remove_ties_mask.numpy()
     if remove_ties_mask.any():
         tie_probs = preds[remove_ties_mask, "Tie"]
         q = (1 - tie_probs)