Update app.py

app.py

@@ -47,7 +47,7 @@ checkpoint = 'wvangils/BLOOM-560m-Beatles-Lyrics-finetuned'
 title_generator = pipeline('summarization', model='czearing/story-to-title')
 lyrics_generator = pipeline("text-generation", model=checkpoint)
 # For the image generator we use stable diffusion from an existing HuggingFace space
-stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
+stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion-2")
 
 # Create 4 images for the given prompt and receive the first one
 # This function uses an existing HuggingFace space where the number of created images cannot be modified