metadata

title: DCGAN MNIST Generator
emoji: 🖼️
colorFrom: blue
colorTo: gray
sdk: gradio
sdk_version: 3.50.0
app_file: app.py
pinned: false

DCGAN MNIST Generator

This repository contains a Deep Convolutional GAN (DCGAN) trained on the MNIST dataset. The model generates handwritten-like digit images from random noise.

Model Architecture

The model implementation is based on the paper Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks.

Generator architecture: 5 transposed convolutional layers with batch normalization
Latent space dimension: 100
Output: 64x64 grayscale images

Demo App

The included Gradio app allows you to generate new MNIST-like images using the pre-trained model.

Running Locally

Install dependencies:

pip install -r requirements.txt