import gradio as gr
from transformers import pipeline

pipe = pipeline("image-classification", model="manoh2f2/finetuned-vit-flowers")
images = ["images/daisy.jpg", "images/dandelion.jpg", "images/rosa.jpg", "images/sunflower.jpg", "images/tulip.jpg"]

iface = gr.Interface.from_pipeline(
        pipe, 
        examples= [ [example] for example in images],
        description="Final project that labels flowers images into: Daisy, Dandelion, Rose, Sunflower, Tulip",
        title="Flower Classifier - Vit"
)

iface.launch()