Added number of bytes for K tokens

app.py

@@ -104,7 +104,7 @@ def infer(img_path, seed=1000, randomize_seed=False, timesteps=25, cfg_scale=7.5
     all_images = [
         (
             TF.to_pil_image(denormalize(reconst_k).clamp(0,1)), 
-            '1 token' if k_keep == 1 else f'{k_keep} tokens'
+            '1 token (2 bytes)' if k_keep == 1 else f'{k_keep} tokens ({2*k_keep} bytes)'
         ) 
         for reconst_k, k_keep in zip(all_reconst, K_KEEP_LIST)
     ]