Spaces:
Sleeping
Sleeping
g and b were switched
Browse files
app.py
CHANGED
|
@@ -12,8 +12,8 @@ learn = load_learner(model_path)
|
|
| 12 |
def classify_bear(img):
|
| 13 |
pred, pred_idx, probs = learn.predict(img)
|
| 14 |
return {
|
| 15 |
-
'
|
| 16 |
-
'
|
| 17 |
'teddy': float(probs[2])
|
| 18 |
}
|
| 19 |
|
|
|
|
| 12 |
def classify_bear(img):
|
| 13 |
pred, pred_idx, probs = learn.predict(img)
|
| 14 |
return {
|
| 15 |
+
'black': float(probs[0]),
|
| 16 |
+
'grizzly': float(probs[1]),
|
| 17 |
'teddy': float(probs[2])
|
| 18 |
}
|
| 19 |
|