diff --git a/README.md b/README.md
index 51565d9..f2bcf2f 100644
--- a/README.md
+++ b/README.md
@@ -9,8 +9,8 @@ This action generate awesome wallpapers using HuggingFace Inference API (serverl
-
-*A geometrically perfect formation of fluorescent jellyfish, floating in the depths of a serene ocean* by [digiplay/insaneRealistic_v1](https://hf.co/digiplay/insaneRealistic_v1)
+
+*A geometrically perfect formation of fluorescent jellyfish, floating in the depths of a serene ocean* by [digiplay/insaneRealistic_v1](https://hf.co/digiplay/insaneRealistic_v1)
@@ -28,11 +28,11 @@ Then add to the [secrets for the repository](https://docs.github.com/en/actions/
## List of models
- [digiplay/insaneRealistic_v1](https://hf.co/digiplay/insaneRealistic_v1)
-- [dreamlike-art/dreamlike-photoreal-2.0](https://hf.co/dreamlike-art/dreamlike-photoreal-2.0)
- [Yntec/DreamPhotoGASM](https://hf.co/Yntec/epiCPhotoGasm)
+- [stabilityai/stable-diffusion-xl-base-1.0](https://hf.co/stabilityai/stable-diffusion-xl-base-1.0)
- [CompVis/stable-diffusion-v1-4](https://hf.co/CompVis/stable-diffusion-v1-4)
-- [Corcelio/mobius](https://hf.co/Corcelio/mobius)
-- [stabilityai/stable-diffusion-2-1](https://hf.co/stabilityai/stable-diffusion-2-1)
+- [prompthero/openjourney-v4](https://hf.co/prompthero/openjourney-v4)
+- [stable-diffusion-v1-5/stable-diffusion-v1-5](https://hf.co/stable-diffusion-v1-5/stable-diffusion-v1-5)
## Outputs
diff --git a/src/generate.py b/src/generate.py
index bb33a0d..a20901d 100644
--- a/src/generate.py
+++ b/src/generate.py
@@ -8,11 +8,11 @@
# The list of text-to-image models that support inference API
MODELS = [
'digiplay/insaneRealistic_v1',
- 'dreamlike-art/dreamlike-photoreal-2.0',
'Yntec/epiCPhotoGasm',
+ 'stabilityai/stable-diffusion-xl-base-1.0',
'CompVis/stable-diffusion-v1-4',
- 'Corcelio/mobius',
- 'stabilityai/stable-diffusion-2-1',
+ 'prompthero/openjourney-v4',
+ 'stable-diffusion-v1-5/stable-diffusion-v1-5',
]
random_model = random.choice(MODELS)