diff --git a/README.md b/README.md
index 1735b09..d5080f8 100644
--- a/README.md
+++ b/README.md
@@ -6,7 +6,7 @@
[](https://discord.gg/xBPBXfcFHd)
-This is a fast, minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini). It has been stripped for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch.
+This is a fast, minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini). It has been stripped down for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch.
To generate a 3x3 grid of DALL·E Mega images it takes
- **35 seconds** with a P100 in Colab
@@ -23,8 +23,6 @@ $ pip install min-dalle
## Usage
-### Python
-
Load the model parameters once and reuse the model to generate multiple images.
```python
@@ -43,11 +41,11 @@ display(image)
```python
-text = 'court sketch of godzilla on trial'
-image = model.generate_image(text, seed=6, grid_size=3)
+text = 'Dali painting of WallE'
+image = model.generate_image(text, seed=0, grid_size=3)
display(image)
```
-
+
```python
text = 'Rusty Iron Man suit found abandoned in the woods being reclaimed by nature'
@@ -56,6 +54,13 @@ display(image)
```
+```python
+text = 'court sketch of godzilla on trial'
+image = model.generate_image(text, seed=6, grid_size=3)
+display(image)
+```
+
+
```python
text = 'a funeral at Whole Foods'
image = model.generate_image(text, seed=10, grid_size=3)
diff --git a/examples/dali_walle.png b/examples/dali_walle.png
new file mode 100644
index 0000000..4ba1fe0
Binary files /dev/null and b/examples/dali_walle.png differ
diff --git a/setup.py b/setup.py
index ec38227..8625142 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,9 @@ setuptools.setup(
license='MIT',
install_requires=[
'torch>=1.10.0',
- 'typing_extensions>=4.1.0'
+ 'typing_extensions>=4.1.0',
+ 'numpy>=1.21',
+ 'pillow>=7.1'
],
keywords = [
'artificial intelligence',