Fixed percentages

app.py

@@ -32,7 +32,10 @@ def predict(img):
   if pred_label_idx:
     pred_labels_and_preds = {class_names[pred_label_idx-1]: float(pred_prob.item())}
   else:
-    pred_labels_and_preds = {class_names[pred_label_idx+1]: 1-float(pred_prob.item())}
+    if 1-float(pred_prob.item()) < 75:
+      pred_labels_and_preds = {class_names[pred_label_idx]: 1-float(pred_prob.item()) }
+    else:
+      pred_labels_and_preds = {class_names[pred_label_idx+1]: 1-float(pred_prob.item())}
 
   end_time = timer()
   pred_time = round(end_time-start_time, 4)