LethallyHealthy commited on
Commit
ade680e
·
verified ·
1 Parent(s): 6f9524a

Attempting to add Pokemon pipeline

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,4 +1,9 @@
1
  import streamlit as st
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
 
 
 
 
 
1
  import streamlit as st
2
+ from diffusers import DiffusionPipeline
3
 
4
+ prompt = st.text_input('Describe the type of pokemon you would like to see:')
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
+ st.image(image)