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 |
|