ju4nppp commited on
Commit
b45e3ee
·
verified ·
1 Parent(s): b6afda1

updated readme

Browse files

corrected readme configuration

Files changed (1) hide show
  1. README.md +32 -42
README.md CHANGED
@@ -1,42 +1,32 @@
1
- # DCGAN MNIST Generator
2
-
3
- This repository contains a Deep Convolutional GAN (DCGAN) trained on the MNIST dataset. The model generates handwritten-like digit images from random noise.
4
-
5
- ## Model Architecture
6
-
7
- The model implementation is based on the paper [Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks](http://arxiv.org/abs/1511.06434).
8
-
9
- - Generator architecture: 5 transposed convolutional layers with batch normalization
10
- - Latent space dimension: 100
11
- - Output: 64x64 grayscale images
12
-
13
- ## Demo App
14
-
15
- The included Gradio app allows you to generate new MNIST-like images using the pre-trained model.
16
-
17
- ### Running Locally
18
-
19
- 1. Install dependencies:
20
- ```bash
21
- pip install -r requirements.txt
22
- ```
23
-
24
- 2. Run the app:
25
- ```bash
26
- python app.py
27
- ```
28
-
29
- ### Features
30
-
31
- - Generate multiple images at once
32
- - Set a random seed for reproducible outputs
33
- - Visualize the generated images in a grid
34
-
35
- ## Training Details
36
-
37
- This model was trained for 25 epochs on the MNIST dataset using PyTorch. For optimal results, the model checkpoint from epoch 21 is used for inference, as it produced the most realistic images without mode collapse.
38
-
39
- ## Acknowledgments
40
-
41
- - Original DCGAN implementation based on [PyTorch examples](https://github.com/pytorch/examples/tree/master/dcgan)
42
- - Training was tracked using Weights & Biases
 
1
+ ---
2
+ title: DCGAN MNIST Generator
3
+ emoji: 🖼️
4
+ colorFrom: blue
5
+ colorTo: gray
6
+ sdk: gradio
7
+ sdk_version: 3.50.0
8
+ app_file: app.py
9
+ pinned: false
10
+ ---
11
+
12
+ # DCGAN MNIST Generator
13
+
14
+ This repository contains a Deep Convolutional GAN (DCGAN) trained on the MNIST dataset. The model generates handwritten-like digit images from random noise.
15
+
16
+ ## Model Architecture
17
+
18
+ The model implementation is based on the paper [Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks](http://arxiv.org/abs/1511.06434).
19
+
20
+ - Generator architecture: 5 transposed convolutional layers with batch normalization
21
+ - Latent space dimension: 100
22
+ - Output: 64x64 grayscale images
23
+
24
+ ## Demo App
25
+
26
+ The included Gradio app allows you to generate new MNIST-like images using the pre-trained model.
27
+
28
+ ### Running Locally
29
+
30
+ 1. Install dependencies:
31
+ ```bash
32
+ pip install -r requirements.txt