jaimetorre commited on
Commit
1d985f4
·
1 Parent(s): b08d721

Added example images. Moved categories before function.

Browse files
Files changed (4) hide show
  1. app.py +1 -3
  2. bear1.jpeg +0 -0
  3. bear2.jpeg +0 -0
  4. bear3.jpeg +0 -0
app.py CHANGED
@@ -7,14 +7,12 @@ warnings.filterwarnings("ignore", category=UserWarning, module="fastai.learner")
7
 
8
  learn = load_learner('export.pkl')
9
 
 
10
 
11
  def classify_bear(image):
12
  pred, pred_idx, probs = learn.predict(image)
13
  return dict(zip(categories, map(float, probs)))
14
 
15
-
16
- categories = ('Black', 'Grizzly', 'Teddy')
17
-
18
  image = gr.Image(height=192, width=192)
19
  label = gr.Label(num_top_classes=3)
20
  examples = ['bear1.jpeg', 'bear2.jpeg', 'bear3.jpeg']
 
7
 
8
  learn = load_learner('export.pkl')
9
 
10
+ categories = ('Black', 'Grizzly', 'Teddy')
11
 
12
  def classify_bear(image):
13
  pred, pred_idx, probs = learn.predict(image)
14
  return dict(zip(categories, map(float, probs)))
15
 
 
 
 
16
  image = gr.Image(height=192, width=192)
17
  label = gr.Label(num_top_classes=3)
18
  examples = ['bear1.jpeg', 'bear2.jpeg', 'bear3.jpeg']
bear1.jpeg ADDED
bear2.jpeg ADDED
bear3.jpeg ADDED