Fix unbiasing

app.py

@@ -30,11 +30,12 @@ def get_interesting_probs(probs, intr_threshold):
 def unbias(p):
     return (2 * p) / (p + 1)
 
+def unbias_or_noop(p, noop=False):
+    return unbias(p) if not noop else p
+
 def predict_bayes(text, intr_threshold, unbiased=False):
     words = tokenize(text)
-    probs = [model_probs.get(w, model_probs[UNK]) for w in words]
-    if unbiased:
-        probs = [unbias(p) for p in probs]
+    probs = [unbias_or_noop(model_probs.get(w, model_probs[UNK]), noop=not unbiased) for w in words]
     interesting_probs = get_interesting_probs(probs, intr_threshold)
     return combine(interesting_probs)