import gradio as gr
from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained("kviai/Kvi-Upscale-V1")

def upscale_image(input_image):
    # Upscale the input image using the diffusion pipeline
    upscaled_image = pipeline.upscale(input_image)
    return upscaled_image

# Create an interface
input_image = gr.Image()  # Input: Image
output_image = gr.Image()  # Output: Image

# Define the Gradio interface to upscale the image
gr.Interface(
    fn=upscale_image,
    inputs=input_image,
    outputs=output_image,
    title="Kvi-Upscaler-V1",
    description="Upload an image to upscale it using Kvi-Upscaler model."
).launch()