typo

app.py

@@ -95,7 +95,10 @@ def infer(img_path, seed=0, randomize_seed=False, timesteps=20, cfg_scale=7.5, p
     
     # Transform to PIL images
     all_images = [
-        (TF.to_pil_image(denormalize(reconst_k).clamp(0,1)), f'{k_keep} tokens') 
+        (
+            TF.to_pil_image(denormalize(reconst_k).clamp(0,1)), 
+            '1 token' if k_keep == 1 else f'{k_keep} tokens'
+        ) 
         for reconst_k, k_keep in zip(all_reconst, K_KEEP_LIST)
     ]