import gradio as gr
from src.classify.classify import classify_logo

if __name__ == "__main__":
    # Create the Gradio interface
    iface = gr.Interface(
        fn=classify_logo,
        inputs=[gr.Image(type="filepath")],
        outputs=gr.Textbox(label="Predicted class")
    )

    # Launch the interface
    iface.launch(share=True)