readme update

buildingamind · Dec 1, 2023 · ded4595 · ded4595
1 parent 19c19ca
commit ded4595
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/README.md b/README.md
@@ -43,15 +43,18 @@ pip install -r requirements.txt
 ## Model Testing
 
 
-## Model Visualization
-
-
 ## Plot Results
 
 <ol>
 <li> <u><b>plotGraphs.ipynb</b></u> - Jupyter notebook containing python code to create bar graphs and line graphs (as shown in the paper). </li>
+
 <br>
 <li> <u><b>visualizeAttentionHeads.ipynb</b></u> - Jupyter notebook containing python code to show the internal representations (saliency maps) for each attention head (as shown in the paper). </li>
+
+<br>
+<p> Example: </p>
+<img src="./media/vis.png" style="height:250px">
+
 </ol>
 
 

diff --git a/media/vis.png b/media/vis.png
diff --git a/notebooks/.DS_Store b/notebooks/.DS_Store