File size: 625 Bytes
205ae77
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import gradio as gr
from transformers import pipeline

pipe = pipeline(task="image-classification", model="SuperSecureHuman/Flower-CNN")

class_names = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']

image = gr.inputs.Image(shape=(300,300))

label = gr.outputs.Label(num_top_classes=5)

gr.Interface.from_pipeline(pipe, 
                           title="Flower Classification",
                           description="Flower CNN",
                           inputs = image,
                           outputs = label,
                           live=True,
                           allow_flagging="never").launch()