Education Diffusion

Inspiration

Teaching and learning deep learning architectures is tough, and visualizations are usually really helpful. Unfortunately, creating visualizations can be tough so we decided to turn to generative AI!

What it does

Uses Stable Diffusion to take in a text prompt and generate an educational image.

How we built it

Finetuned Stable Diffusion based on a dataset we created by web crawling for educational diagrams and images.

Challenges we ran into

Hosting on a server, using a pre-trained model, and generating a powerful dataset was much tougher than we expected.

Accomplishments that we're proud of

Getting the model hosted on a server that's super fast and epic!

What we learned

Many packages advertise themselves as plug-and-play, but that is rarely the case sadly.

What's Next for Education Diffusion

We'd like to train larger models on more data, and potentially extend to more fields of education

Built With

• claude
• google-cloud
• python
• pytorch
• stable-diffusion