import gradio as gr
# Load the RunwayML model using Gradio
model = gr.load("models/runwayml/stable-diffusion-v1-5")
# Custom HTML and CSS for styling
custom_html = """
Stable Diffusion Model
{input}
{output}
"""
# Launch the interface with the loaded model and custom styling
iface = gr.Interface(fn=model, inputs=model.interface.inputs, outputs=model.interface.outputs)
iface.share(html=custom_html)