Spaces:
Runtime error
Runtime error
File size: 1,060 Bytes
bd74121 4d83aec f2241ff 600e525 4d83aec f2241ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import streamlit as st
import requests
from PIL import Image
from io import BytesIO
st.title('CreativeAI')
# Description
st.write("""
This app uses HuggingFace to generate an image based on the provided prompt. You can either select an example prompt from the dropdown list or enter your own prompt.
Click the "Generate Image" button to initiate the image generation process. Based on KVIImager API by KVI Kontent.
""")
custom_prompt = st.text_input("Enter your own prompt")
prompt = custom_prompt
if st.button('Generate Image'):
if prompt:
with st.spinner('Generating your image...'):
url = 'https://ee1f-217-71-237-228.ngrok-free.app/generate_image'
params = {'prompt': prompt}
response = requests.get(url, params=params)
if response.status_code == 200:
image = Image.open(BytesIO(response.content))
st.image(image, caption='Generated Image', use_column_width=True)
else:
st.error("Failed to generate the image. Please try again.")
|