File size: 1,886 Bytes
6ee020d
1084b0c
6ee020d
f864afc
 
6ee020d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61fa516
6ee020d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1d660f9
 
 
 
6ee020d
 
 
61fa516
6ee020d
 
 
 
 
61fa516
 
6ee020d
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
---
title: Custom Pokemon Generator
emoji: 🎨
colorFrom: red
colorTo: pink
sdk: gradio
sdk_version: 5.25.2
app_file: app.py
pinned: true
license: mit
short_description: Generate unique Pokémon-style character illustrations!
---

# 🧬 Pokémon Image Generator with Stable Diffusion 🎨

This is a Gradio web application that generates Pokémon-style character images from text prompts.#

## 📌 What It Does

- 🔥 Takes a text prompt (e.g. "fire fox with blue flames")
- 🎨 Generates an original Pokémon-style creature

🧠 Model Details

This project uses a fine-tuned version of the Stable Diffusion v1.5 model, originally based on [`runwayml/stable-diffusion-v1-5`](https://huggingface.co/runwayml/stable-diffusion-v1-5) (now offline).  
The fine-tuned model is available here:  
👉 [`nikkijiang/sd-pokemon-generator`](https://huggingface.co/nikkijiang/sd-pokemon-generator)


## 📚 Training Datasets

The model was fine-tuned using image-caption datasets of Pokémon-style creatures:

- [lambdalabs/pokemon-blip-captions](https://huggingface.co/datasets/lambdalabs/pokemon-blip-captions)  
- [42arch/pokemon-dataset-zh](https://github.com/42arch/pokemon-dataset-zh)

These datasets include curated images of Pokémon-like monsters with captions, enabling text-to-image training in a stylized domain.

## 🖼️ Example Prompts

Try prompts like:

- `lovely cat, wings, shining`
- `flower dog, big eyes, long ear`
- `cute, rabbit, pink, attack`
- `bird with red hat and coat, real`

## 🌐 Online Demo

👉 [Try it on Hugging Face Spaces](https://huggingface.co/spaces/nikkijiang/custom-pokemon-generator)  
Built with [Gradio](https://gradio.app/) for a clean, interactive interface.

## 🚀 Run Locally

```bash
git clone https://huggingface.co/spaces/nikkijiang/custom-pokemon-generator
cd custom-pokemon-generator
pip install -r requirements.txt
python app.py