File size: 992 Bytes
09fb9f0 5941390 d127695 5941390 09fb9f0 d127695 5941390 d127695 66eaf5f d127695 66eaf5f d127695 66eaf5f 5941390 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import gradio as gr
import spaces
from gradio_imageslider import ImageSlider
from image_gen_aux import UpscaleWithModel
from image_gen_aux.utils import load_image
@spaces.GPU
def upscale_image(image):
original = load_image(image)
upscaler = UpscaleWithModel.from_pretrained("OzzyGT/UltraSharp").to("cuda")
image = upscaler(original)
return original, image
title = """<h1 align="center">Image Upscaler</h1>"""
with gr.Blocks() as demo:
gr.HTML(title)
with gr.Row():
with gr.Column():
input_image = gr.Image(type="pil", label="Input Image")
run_button = gr.Button("Upscale")
with gr.Column():
result = ImageSlider(
interactive=False,
label="Generated Image",
elem_id="result-image",
position=0.1,
)
run_button.click(
fn=upscale_image,
inputs=[input_image],
outputs=result,
)
demo.launch(share=False)
|