Low-light / app.py
Brian Gabini
feat: change image upload method
71bcdff
raw
history blame
1.15 kB
import gradio as gr
# project
from exposure_enhancement import enhance_image_exposure
# inputs, fn, and ouputs
examples=[
["demo/1.jpg"],
["demo/2.bmp"]
]
def enhance_image(image, gamma=0.6, lambda_=0.15, sigma=3, lime=True, bc=1, bs=1, be=1, eps=1e-3):
# enhance image
enhanced_image = enhance_image_exposure(image, gamma, lambda_, not lime, sigma=sigma, bc=bc, bs=bs, be=be, eps=eps)
return enhanced_image
def update(name):
return f"Welcome to Gradio, {name}!"
with gr.Blocks() as demo:
gr.Markdown("Input an image below then click **Run** or use the cached examples for quick results. For quicker processing (typically 1-2 minutes), it's suggested to use low-exposure images of 500x500 pixels.")
with gr.Row():
inp = gr.Image(label="Input Image", type="numpy", sources=["upload"])
out = gr.Image(label="Enhanced Image", type="numpy")
with gr.Row():
btn = gr.Button("Run")
btn.click(fn=enhance_image, inputs=inp, outputs=out)
btn = gr.ClearButton()
with gr.Row():
gr.Examples(examples=examples, inputs=inp)
demo.launch(share=True)