manoh2f2 commited on
Commit
a0d03a0
·
1 Parent(s): 6555183

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -4,12 +4,10 @@ model = gr.load("models/manoh2f2/finetuned-vit-flowers")
4
  images = ["images/daisy.jpg", "images/dandelion.jpg", "images/rosa.jpg", "images/sunflower.jpg", "images/tulip.jpg"]
5
 
6
  iface = gr.Interface(
7
- fn = model,
8
- inputs = gr.Image(type="pil", label="Upload Image"),
9
- outputs = gr.Textbox(),
10
- live = True,
11
- examples = images,
12
- title = "Flower Classifier"
13
  )
14
 
15
  iface.launch()
 
4
  images = ["images/daisy.jpg", "images/dandelion.jpg", "images/rosa.jpg", "images/sunflower.jpg", "images/tulip.jpg"]
5
 
6
  iface = gr.Interface(
7
+ model,
8
+ examples= [ [example] for example in images],
9
+ description="Daisy, Dandelion, Rose, Sonflower, Tulip",
10
+ title="FLower Classifier - Vit"
 
 
11
  )
12
 
13
  iface.launch()