:tada: init

app.py

@@ -301,7 +301,10 @@ def inference(
             attn_probs = cross_attention_probs[i, j].cpu().detach().numpy()
             ax[j].imshow(attn_probs, alpha=0.9)
             ax[j].axis('off')
-            ax[j].set_title(tokenizer.decode(input_ids[0, j].item()))
+            if has_include_special_tokens:
+                ax[j].set_title(tokenizer.decode(input_ids[0, j].item()))
+            else:
+                ax[j].set_title(tokenizer.decode(input_ids[0, j + 1].item()))
 
         # 各行ごとの画像を保存
         out_dir = Path("output")