mnoorchenar commited on
Commit
a360e21
·
verified ·
1 Parent(s): 5d3e724

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -26
README.md CHANGED
@@ -1,28 +1,10 @@
1
- # 🧠 Variational Autoencoder (VAE) Interactive Playground
2
-
3
- An interactive Flask web app that lets you **train, visualize, and generate MNIST digits** using a Variational Autoencoder (VAE).
4
- Built with **PyTorch**, **Flask**, and **Matplotlib**.
5
-
6
  ---
7
-
8
- ## 🚀 Features
9
- - Train a VAE model directly in your browser
10
- - Real-time loss visualization
11
- - Latent space plotting (2D)
12
- - Reconstruction and generation of handwritten digits
13
- - Adjustable architecture and hyperparameters
14
-
15
  ---
16
-
17
- ## 🧩 Tech Stack
18
- - **Flask** for the web interface
19
- - **PyTorch** for model training
20
- - **Matplotlib** for plots
21
- - **Threading** for non-blocking training
22
-
23
- ---
24
-
25
- ## ⚙️ Run locally
26
- ```bash
27
- pip install -r requirements.txt
28
- python app.py
 
 
 
 
 
 
1
  ---
2
+ title: VAE Interactive Playground
3
+ emoji: 🧠
4
+ colorFrom: indigo
5
+ colorTo: purple
6
+ sdk: static
7
+ sdk_version: "1.0.0"
8
+ app_file: app.py
9
+ pinned: false
10
  ---