namanhboi commited on
Commit
0399aa2
1 Parent(s): 11bcdb8

finally i think

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -4,14 +4,14 @@ from fastai.vision.all import *
4
  learn = load_learner("export.pkl")
5
  examples = ['jokicpassing.jpg', 'modric.jpg', 'tennis.jpg']
6
 
7
- categories = ('tennis', 'soccer', 'basketball')
8
 
9
  def classify_image(img):
10
  pred,idx,prob = learn.predict(img)
11
- return dict(zip(categories), map(float, prob))
12
 
13
- images = gr.inputs.Image(shape = (224, 224))
14
- labels = gr.outputs.Label()
15
 
16
  intf = gr.Interface(fn = classify_image, inputs=images, outputs=labels, examples=examples)
17
  intf.launch(inline = False)
 
4
  learn = load_learner("export.pkl")
5
  examples = ['jokicpassing.jpg', 'modric.jpg', 'tennis.jpg']
6
 
7
+ categories = ('basketball', 'soccer', 'tennis')
8
 
9
  def classify_image(img):
10
  pred,idx,prob = learn.predict(img)
11
+ return dict(zip(categories, map(float, prob)))
12
 
13
+ images = gr.Image(height = 224, width=224)
14
+ labels = gr.Label()
15
 
16
  intf = gr.Interface(fn = classify_image, inputs=images, outputs=labels, examples=examples)
17
  intf.launch(inline = False)
requirements.txt CHANGED
@@ -0,0 +1 @@
 
 
1
+ fastai=2.7.12