samcoding5854 commited on
Commit
163abe0
1 Parent(s): f0a1f4f

adjusted image sizes

Browse files
Files changed (1) hide show
  1. Pages/bgImages.py +2 -2
Pages/bgImages.py CHANGED
@@ -31,7 +31,7 @@ def BGIMAGES():
31
  # Generate the image
32
  with st.spinner("Generating image..."):
33
  image_prompt = f"{prompt}, muted colors, detailed, 8k"
34
- image = pipeline(image_prompt, generator=generator).images[0]
35
 
36
  # Save the image
37
  output_dir = "Assets/bgImages"
@@ -40,7 +40,7 @@ def BGIMAGES():
40
  image.save(image_path)
41
 
42
  # Display the image
43
- st.image(image, caption=f"Generated image for: {prompt}")
44
 
45
  else:
46
  st.write("Please enter a movie title to generate an image.")
 
31
  # Generate the image
32
  with st.spinner("Generating image..."):
33
  image_prompt = f"{prompt}, muted colors, detailed, 8k"
34
+ image = pipeline(image_prompt, generator=generator, height=512, width=768).images[0]
35
 
36
  # Save the image
37
  output_dir = "Assets/bgImages"
 
40
  image.save(image_path)
41
 
42
  # Display the image
43
+ st.image(image, caption=f"Generated image for: {prompt}", width = 300)
44
 
45
  else:
46
  st.write("Please enter a movie title to generate an image.")