manoh2f2 commited on
Commit
d3ba833
1 Parent(s): 73ee263

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -1,3 +1,14 @@
1
  import gradio as gr
2
 
3
- gr.load("models/manoh2f2/finetuned-vit-flowers").launch()
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ 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", multiple=True),
9
+ outputs = gr.Textbox(),
10
+ live = True,
11
+ examples = [ [example] for example in images]
12
+ )
13
+
14
+ iface.launch()