import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
import warnings
warnings.filterwarnings('ignore')
import gradio as gr
from PIL import Image
from test_simple import test_simple

def predict(image: Image):
    return test_simple(image)

title = "Underwater Image Restoration"

iface = gr.Interface(
    predict,
    inputs=gr.Image(type="pil"),
    outputs="image",
    title=title,
    allow_flagging="never",
    examples=[
        ["flatiron_1.tiff"],
        ["flatiron_2.tiff"],
        ["horse_canyon_1.tiff"],
        ["horse_canyon_2.tiff"],
    ],
)
iface.launch(share=True)