Spaces:
Running
Running
| import streamlit as st | |
| # import openai | |
| import replicate | |
| def page2(): | |
| st.title("OpenAI DALL·E Image Generation") | |
| st.info("""#### NOTE: you can download image by \ | |
| right clicking on the image and select save image as option""") | |
| with st.form(key='form'): | |
| prompt = st.text_input(label='Enter text prompt for image generation') | |
| size = st.selectbox('Select size of the images', | |
| ('256x256', '512x512', '1024x1024')) | |
| num_images = st.selectbox('Enter number of images to be generated', (1,2,3,4)) | |
| submit_button = st.form_submit_button(label='Submit') | |
| if submit_button: | |
| if prompt: | |
| output = replicate.run( | |
| "stability-ai/stable-video-diffusion:3f0457e4619daac51203dedb472816fd4af51f3149fa7a9e0b5ffcf1b8172438", | |
| input={ | |
| "input_image": "https://example.com/path/to/file/input_image" | |
| } | |
| ) | |
| print(output) | |
| # response = openai.Image.create( | |
| # prompt = prompt, | |
| # n = num_images, | |
| # size=size, | |
| # ) | |
| # for idx in range(num_images): | |
| # image_url = response["data"][idx]["url"] | |
| # st.image(image_url, caption=f"Generated image: {idx+1}", | |
| # use_column_width=True) |