keplersj commited on
Commit
7d3d31c
·
1 Parent(s): 6678a82

ui goodies

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,14 +10,14 @@ pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
10
  captions = []
11
 
12
  with st.sidebar:
 
 
13
  image_gen_guidance = st.slider("Stable Diffusion: Guidance Scale", value=7.5)
14
  image_gen_steps = st.slider("stable Diffusion: Inference Steps", value=50)
15
 
16
  col1, col2 = st.columns(2)
17
 
18
  with col1:
19
- files = st.file_uploader("Upload images to blend", accept_multiple_files=True)
20
-
21
  for file_name in files:
22
  image = Image.open(file_name)
23
 
@@ -33,7 +33,7 @@ with col2:
33
  if len(captions) > 0:
34
  description = ' '.join(captions)
35
 
36
- with st.spinner('Generating Photo from Caption'):
37
  images = pipe(description, guidance_scale=image_gen_guidance, num_inference_steps=image_gen_steps).images
38
 
39
  for image in images:
 
10
  captions = []
11
 
12
  with st.sidebar:
13
+ files = st.file_uploader("Upload images to blend", accept_multiple_files=True)
14
+ st.divider()
15
  image_gen_guidance = st.slider("Stable Diffusion: Guidance Scale", value=7.5)
16
  image_gen_steps = st.slider("stable Diffusion: Inference Steps", value=50)
17
 
18
  col1, col2 = st.columns(2)
19
 
20
  with col1:
 
 
21
  for file_name in files:
22
  image = Image.open(file_name)
23
 
 
33
  if len(captions) > 0:
34
  description = ' '.join(captions)
35
 
36
+ with st.spinner(f'Generating Photo for {description}'):
37
  images = pipe(description, guidance_scale=image_gen_guidance, num_inference_steps=image_gen_steps).images
38
 
39
  for image in images: