arham061 commited on
Commit
12197c3
1 Parent(s): a31c781

Add application file

Browse files
Files changed (1) hide show
  1. rps.py +2 -2
rps.py CHANGED
@@ -9,7 +9,7 @@ learn = load_learner('rps_model.pkl')
9
 
10
  categories = ('paper', 'rock', 'scissors')
11
 
12
- def classify_image(img):
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
@@ -17,5 +17,5 @@ image = gr.inputs.Image(shape=(192,192))
17
  label = gr.outputs.Label()
18
  examples = ['rock.jpg', 'paper.jpg', 'scissors.jpg']
19
 
20
- intf = gr.Interface(fn = classify_image, inputs = image, outputs = label, examples = examples)
21
  intf.launch(inline=False)
 
9
 
10
  categories = ('paper', 'rock', 'scissors')
11
 
12
+ def classify_images(img):
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
 
17
  label = gr.outputs.Label()
18
  examples = ['rock.jpg', 'paper.jpg', 'scissors.jpg']
19
 
20
+ intf = gr.Interface(fn = classify_images, inputs = image, outputs = label, examples = examples)
21
  intf.launch(inline=False)