Spaces:
Sleeping
Sleeping
Commit
·
a095fd1
1
Parent(s):
f3fb2e1
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def predict(img):
|
|
| 37 |
pred_and_prob_labels = {class_names[i]: preds[0][i].item() for i in range(len(class_names))}
|
| 38 |
|
| 39 |
else:
|
| 40 |
-
pred_and_prob_labels = {"Low Accuracy Warning !!! Kindly verify whether the given image is a
|
| 41 |
|
| 42 |
return pred_and_prob_labels
|
| 43 |
|
|
|
|
| 37 |
pred_and_prob_labels = {class_names[i]: preds[0][i].item() for i in range(len(class_names))}
|
| 38 |
|
| 39 |
else:
|
| 40 |
+
pred_and_prob_labels = {"Low Accuracy Warning !!! Kindly verify whether the given image is an image of a bird, or upload a clear image.": preds[0].max().item()}
|
| 41 |
|
| 42 |
return pred_and_prob_labels
|
| 43 |
|