import gradio as gr
from transformers import pipeline


pipe = pipeline(task="image-classification", model="rexoscare/string_instrument_detector")


gr.Interface.from_pipeline(pipe,
                           title="Stringed Instrument Classification",
                           description="It will predict the Stringed Instrument with accuracy",
                           examples = ["guitar.jpg", "uke.jpg",],
                           allow_flagging="never"
                           ).launch(inbrowser=True)