Spaces:
Sleeping
Sleeping
import gradio as gr | |
from PIL import Image | |
from utils import generate_gradio_images | |
# Create a Gradio interface | |
iface = gr.Interface( | |
fn=generate_gradio_images, # Set to None since we won't use a function | |
inputs=[ | |
gr.Textbox("Enter Prompt"), | |
gr.Slider("Time Stamps", value=1, step=1, maximum=50), | |
gr.Checkbox(label = "Use custom loss",value=False) | |
], | |
outputs=[gr.Image(type="pil", height=128, width = 128), | |
gr.Image(type="pil", height=128, width = 128), | |
gr.Image(type="pil", height=128, width = 128), | |
gr.Image(type="pil", height=128, width = 128), | |
gr.Image(type="pil", height=128, width = 128)] # Gradio Image component for displaying images | |
) | |
# Launch the Gradio interface | |
iface.launch() |