Update README.md

README.md

@@ -3,8 +3,6 @@
 [![Build Status](https://travis-ci.org/user/repo.svg?branch=master)](https://travis-ci.org/user/repo)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
 
-![Alt text](Attention.drawio.png)
-
 ## Overview
 
 This library is designed to generate visualizations of neural networks by analyzing PyTorch code. Unlike conventional methods, our approach eliminates the need for large parameter files (pth) or specific input formats. With minimal AI knowledge, users can leverage this tool to create visualizations that aid in understanding complex networks, including support for transformers and attention mechanisms.
@@ -120,23 +118,25 @@ This project is licensed under the [MIT License](https://opensource.org/licenses
 
 ## Project Status
 
-Active development status or any relevant project status updates.
-
 ## Community and Contact
 
-Engage with the community on our [forum](https://community-forum-link.com) or contact us via email at [your-email@example.com].
-
-## Badges
-
-Display badges indicating project status, build, version, and other relevant information.
+Contact us via email at [yibao@ucsd.edu].
 
 ## Screenshots or Demos
 
-Include screenshots or links to demos showcasing the visualizations generated by the library.
+![Alt text](Attention.drawio.png)
 
 ## Related Projects
 
-Explore other projects or tools related to neural network visualization.
+- https://github.com/gwding/draw_convnet
+- http://alexlenail.me/NN-SVG/LeNet.html
+- https://github.com/HarisIqbal88/PlotNeuralNet
+- https://www.tensorflow.org/tensorboard/graphs
+- https://github.com/BVLC/caffe/blob/master/python/caffe/draw.py
+- https://www.mathworks.com/help/nnet/ref/view.html
+- https://transcranial.github.io/keras-js/%23/inception-v3
+- https://github.com/stared/keras-sequential-ascii/
+- https://github.com/lutzroeder/Netron
 
 ## Frequently Asked Questions (FAQ)