import gradio as gr

model = gr.load("models/manoh2f2/finetuned-vit-flowers")
images = ["images/daisy.jpg", "images/dandelion.jpg", "images/rosa.jpg", "images/sunflower.jpg", "images/tulip.jpg"]

iface = gr.Interface(
        model, 
        inputs = gr.Image(label="Upload Image"),
        outputs = gr.Textbox(),
        examples= [ [example] for example in images],
        description="Daisy, Dandelion, Rose, Sonflower, Tulip",
        title="FLower Classifier - Vit"
)

iface.launch()