Shreyavalte commited on
Commit
13874cd
1 Parent(s): 246e10c

Update app.py

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