Englios commited on
Commit
09dc58d
1 Parent(s): 73c32dc

Fix app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,10 +1,12 @@
1
- __all__ = ['learn','categories','classify_images','image','label','examples','intf']
2
 
3
  import gradio as gr
4
  from fastai.vision.all import *
5
 
 
 
6
 
7
- learn = load_learner('model.pkl')
8
 
9
  categories = ('Grizzly','Black','Toy','White','Brown')
10
 
@@ -14,7 +16,7 @@ def classify_image(img):
14
 
15
  image = gr.Image(sources="upload")
16
  label = gr.Label()
17
- examples = ['black.jpg','brown.jpg','grizzly.jpeg','toy.jpeg','./Imgs/white.jpg']
18
 
19
  intf = gr.Interface(fn=classify_image,inputs=image,outputs=label,examples=examples)
20
  intf.launch(inline=False,share=True)
 
1
+ __all__ = ['getx','gety','learn','categories','classify_images','image','label','examples','intf']
2
 
3
  import gradio as gr
4
  from fastai.vision.all import *
5
 
6
+ def getx(r):return r['fname']
7
+ def gety(r):return r['cat'].split(' ')
8
 
9
+ learn = load_learner('./model.pkl')
10
 
11
  categories = ('Grizzly','Black','Toy','White','Brown')
12
 
 
16
 
17
  image = gr.Image(sources="upload")
18
  label = gr.Label()
19
+ examples = ['black.jpg','brown.jpg','grizzly.jpeg','toy.jpeg','white.jpg']
20
 
21
  intf = gr.Interface(fn=classify_image,inputs=image,outputs=label,examples=examples)
22
  intf.launch(inline=False,share=True)