S20_EraV1_SD / app.py
Vasudevakrishna's picture
Update app.py
3eacd6a verified
raw
history blame contribute delete
785 Bytes
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=512, width = 512),
gr.Image(type="pil", height=512, width = 512),
gr.Image(type="pil", height=512, width = 512),
gr.Image(type="pil", height=512, width = 512),
gr.Image(type="pil", height=512, width = 512)] # Gradio Image component for displaying images
)
# Launch the Gradio interface
iface.launch()