Kathir0011 commited on
Commit
a095fd1
·
1 Parent(s): f3fb2e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 Bird.": preds[0].max().item()}
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