Abdullah-Habib commited on
Commit
246b3a7
·
1 Parent(s): 645ab3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,12 +3,12 @@ __all__ = ['is_Rabbit',"learn",'classify_image', 'categories','image','label','e
3
  # Cell
4
  from fastai.vision.all import *
5
  import gradio as gr
6
- def is_cat(x): return x[1].isupper()
7
 
8
 
9
  learn = load_learner ('model.pkl')
10
  # Cell
11
- categories = ('Rabbit', 'Hare')
12
  def classify_image (img) :
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float,probs)))
 
3
  # Cell
4
  from fastai.vision.all import *
5
  import gradio as gr
6
+ def is_Rabbit(x): return x[0].isupper()
7
 
8
 
9
  learn = load_learner ('model.pkl')
10
  # Cell
11
+ categories = ('Hare','Rabbit')
12
  def classify_image (img) :
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float,probs)))