File size: 1,160 Bytes
1a548a3 |
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 |
import os
import gradio as gr
from synthesizer import SRSynthesizer
from gradio_imageslider import ImageSlider
if __name__ == "__main__":
sr_synthesizer = SRSynthesizer(create_dirs=False)
gr_interface = gr.Interface(
fn=lambda image: sr_synthesizer.synthesize(image,
show=False,
save=False,
return_input=True),
inputs=[gr.Image(type="pil", label="Input")],
outputs=ImageSlider(type="pil", label="Output", show_download_button=True),
title="Super Resolution Image Synthesizer",
examples=[
[os.path.join(os.path.dirname(__file__), "low-res-images", "building.png")],
[os.path.join(os.path.dirname(__file__), "low-res-images", "plant.png")],
[os.path.join(os.path.dirname(__file__), "low-res-images", "penguin.png")],
[os.path.join(os.path.dirname(__file__), "low-res-images", "vietnam_park.jpg")],
],
description="Synthesize (4x-upscaled) super-resolved images"
)
gr_interface.launch()
|