3laa2 commited on
Commit
cd574a7
·
1 Parent(s): a0b3ec0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,8 +7,8 @@ import cv2 as cv
7
  st.title("Txt2Img")
8
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16")
9
  pipe.to("cpu")
10
- area = st.text_area("Area",'')
11
- if len(area)>5:
12
- image = pipe(area).images[0]
13
  st.image(image)
14
 
 
7
  st.title("Txt2Img")
8
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16")
9
  pipe.to("cpu")
10
+ prom = st.text_area("Prompt",'')
11
+ if len(prom)>5:
12
+ image = pipe(prom).images[0]
13
  st.image(image)
14