Yousef Rafat commited on
Commit
bf02e71
·
1 Parent(s): 0302379

Corrected Labels

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. app.py +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
- title: Cat or Dog Classifier
3
- emoji: 🦁
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
 
1
  ---
2
+ title: Fungi | Poisonous or Healthy
3
+ emoji: 🍄
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
app.py CHANGED
@@ -11,7 +11,7 @@ def is_cat(x): return x[0].isupper()
11
 
12
  learn = load_learner('my_model.pkl')
13
 
14
- categories = ('Dog', 'Cat')
15
 
16
  def classify_image(img):
17
  pred, idx, probs = learn.predict(img)
 
11
 
12
  learn = load_learner('my_model.pkl')
13
 
14
+ categories = ('Healthy Fungi', 'Poisonous Fungi')
15
 
16
  def classify_image(img):
17
  pred, idx, probs = learn.predict(img)