File size: 490 Bytes
f592388
 
d3ba833
 
 
 
a0d03a0
bfa5aa7
f1d8420
a0d03a0
 
 
d3ba833
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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()