Update app.py

app.py

@@ -4,6 +4,7 @@ from diffusers import DiffusionPipeline
 prompt = st.text_input('Describe the type of pokemon you would like to see:')
 if(prompt):
     st.spinner(text='Generating your pokemon...')
-pipe = DiffusionPipeline.from_pretrained("justinpinkney/pokemon-stable-diffusion")
-image = pipe(prompt).images[0]
-st.image(image)
+    pipe = DiffusionPipeline.from_pretrained("justinpinkney/pokemon-stable-diffusion")
+    image = pipe(prompt).images[0]
+if(image): 
+    st.image(image)