Spaces:
Runtime error
Runtime error
Yousef Rafat
commited on
Commit
·
bf02e71
1
Parent(s):
0302379
Corrected Labels
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
-
title:
|
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 = ('
|
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)
|