Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
import gradio as gr | |
import kornia as K | |
from kornia.core import Tensor | |
def enhance(file): | |
# load the image using the rust backend | |
img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32) | |
img = img[None] # 1xCxHxW / fp32 / [0, 1] | |
return K.utils.tensor_to_image(img) | |
examples = [ | |
['examples/foto1B.jpg'], | |
['examples/foto1A.jpg'], | |
] | |
iface = gr.Interface( | |
enhance, | |
[ | |
gr.inputs.Image(type="file"), | |
], | |
"image", | |
examples=examples, | |
# title=title, | |
# description=description, | |
# article=article, | |
live=True | |
) | |
iface.launch() |