Update app.py

app.py

@@ -117,15 +117,15 @@ def infer(model, replay_file,
     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"]
+        tie_probs = preds.loc[remove_ties_mask, "Tie"]
         q = (1 - tie_probs)
         for c in preds.columns:
             if c.startswith("Blue") or c.startswith("Orange"):
-                preds[remove_ties_mask, c] /= q
+                preds.loc[remove_ties_mask, c] /= q
         if ignore_ties:
             preds = preds.drop("Tie", axis=1)
         else:
-            preds[remove_ties_mask, "Tie"] = 0.0
+            preds.loc[remove_ties_mask, "Tie"] = 0.0
 
     return preds