ishaal007 commited on
Commit
3eeff5e
·
1 Parent(s): bd61261

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def predict(img) -> Tuple[Dict, float]:
53
  x=np.count_nonzero(pred_prob)
54
  print(x)
55
  if x==0:
56
- pred_labels_and_probs={"Unknown Images found":0}
57
  else:
58
  pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
59
 
 
53
  x=np.count_nonzero(pred_prob)
54
  print(x)
55
  if x==0:
56
+ pred_labels_and_probs={"Unknown Images found...Please Provide Images of Headphone,Mobile or laptop":0}
57
  else:
58
  pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
59