khan994 commited on
Commit
a10cf7f
·
1 Parent(s): 4a903b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
 
3
 
4
  class Hook():
5
  def hook_func(self, m, i, o): self.stored = o.detach().clone()
@@ -13,11 +14,15 @@ def classify_img(img):
13
 
14
  image=gr.inputs.Image(shape=(128,128))
15
  label=gr.outputs.Label()
 
 
 
 
16
  examples=["filibe-1-1.jpg",
17
  "ohrid-3-1.jpg",
18
  "varna-1-1.jpg"]
19
 
20
 
21
- demo = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
22
 
23
  demo.launch(inline=False)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
+ import glob
4
 
5
  class Hook():
6
  def hook_func(self, m, i, o): self.stored = o.detach().clone()
 
14
 
15
  image=gr.inputs.Image(shape=(128,128))
16
  label=gr.outputs.Label()
17
+ examples_=[]
18
+ for i in glob.glob("/valid/**/*.jpg", recursive=True):
19
+ examples_.append(i)
20
+ examples_
21
  examples=["filibe-1-1.jpg",
22
  "ohrid-3-1.jpg",
23
  "varna-1-1.jpg"]
24
 
25
 
26
+ demo = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples_)
27
 
28
  demo.launch(inline=False)