Spaces:
Running
Running
samcoding5854
commited on
Commit
•
163abe0
1
Parent(s):
f0a1f4f
adjusted image sizes
Browse files- 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.")
|