manoh2f2's picture
Update app.py
bfa5aa7
raw
history blame
490 Bytes
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()