import gradio as gr
from predict import predict

demo = gr.Interface(
    fn=predict,
    inputs=gr.Image(type="filepath", label="Upload Image"),
    outputs=[
        gr.Image(label="Uploaded Image"),
        gr.Image(label="Top-1 Class Example"),
        gr.Label(label="Top-5 Probabilities")
    ],
    title="Scene Classification with Reference Image Testing using Large SWIN version 1.1.0",
    description="Upload an image to get the predicted class with a sample image and top-5 prediction chart."
)

demo.launch()