import gradio as gr
from inference import predict

interface = gr.Interface(
    fn=predict,
    inputs=gr.Image(type="filepath", label="Upload SAR Image"),
    outputs="image",
    title="SAR Image Colorization"
)

interface.launch()