Update app.py

app.py

@@ -116,8 +116,11 @@ def infer(model, replay_file,
     preds.index.name = "Frame"
     if ignore_ties:
         tie_probs = preds["Tie"]
+        q = (1 - tie_probs)
         preds = preds.drop("Tie", axis=1)
-        preds[[c for c in preds.columns if c.startswith("Blue") or c.startswith("Orange")]] /= (1 - tie_probs)
+        for c in preds.columns:
+            if c.startswith("Blue") or c.startswith("Orange"):
+                preds[c] /= q
 
     return preds