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) |