image-upscaler / app.py
ehristoforu's picture
Update app.py
bf31677
raw
history blame
658 Bytes
import gradio as gr
import cv2
import numpy as np
def upscale_image(input_image):
# Увеличиваем размер изображения в 2 раза
upscale_factor = radio_input
output_image = cv2.resize(input_image, None, fx = upscale_factor, fy = upscale_factor, interpolation = cv2.INTER_CUBIC)
return output_image
radio_input = gr.Radio(label="Upscale Levels", choices=["2", "4"], value="2")
# Создаем интерфейс Gradio
iface = gr.Interface(fn=upscale_image, inputs = [gr.Image(label="Input Image", interactive=True), radio_input], outputs = gr.Image(label="Upscaled Image"), title="Image Upscaler")
iface.launch()