Draw a pytorch model app with streamlit

An app for you to draw a diagram of your pytorch model and get the code for it


This is initial stage. Help, contributions and feedback are greatly appreciated


Hi @and_b,

Your app is pretty cool! Thanks so much for sharing!

Happy Streamlit-ing

Hi @and_b. I was trying to use the app.
Would it be possible to keep some sample images on-site so that users could get an idea about the expected artwork?

Thank you for trying out. I put a video demo here