Update app.py

app.py

@@ -17,7 +17,6 @@ import gradio as gr
 import re
 import torch
 from torch import autocast
-from PIL import Image
 import os
 
 # Array with song cover art styles
@@ -48,7 +47,7 @@ checkpoint = 'wvangils/GPT-Medium-Beatles-Lyrics-finetuned-newlyrics'
 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-1")
+stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
 
 # 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