LethallyHealthy commited on
Commit
7da94b5
·
verified ·
1 Parent(s): 14cd75d

Update app.py

Browse files

changed scope of image

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,5 +6,5 @@ if(prompt):
6
  st.spinner(text='Generating your pokemon...')
7
  pipe = DiffusionPipeline.from_pretrained("justinpinkney/pokemon-stable-diffusion")
8
  image = pipe(prompt).images[0]
9
- if(image):
10
- st.image(image)
 
6
  st.spinner(text='Generating your pokemon...')
7
  pipe = DiffusionPipeline.from_pretrained("justinpinkney/pokemon-stable-diffusion")
8
  image = pipe(prompt).images[0]
9
+ if(image):
10
+ st.image(image)