fix

app.py

@@ -86,11 +86,12 @@ def inference(
     lm_tokenizer = transformers.AutoTokenizer.from_pretrained(lm_model_name)
     if alpha != 0:
         caif_sampler = load_sampler(cls_model_name=cls_model_name, lm_tokenizer=lm_tokenizer)
+        if entropy_threshold < 0.05:
+            entropy_threshold = None
     else:
         caif_sampler = None
         entropy_threshold = None
-    if entropy_threshold < 0.05:
-        entropy_threshold = None
+
     generator.set_caif_sampler(caif_sampler)
     ordinary_sampler = TopKWithTemperatureSampler()
     kwargs = {