Fix probability threshold checking logic

app.py

@@ -10,10 +10,13 @@ def classify_image(img):
     pred, idx, probs = learn.predict(img)
     output = dict(zip(labels, map(float, probs)))
     print(probs[0].item()*100)
+    count = 0
     for out in output:
         val = output[out]
         if val*100 < 60:
-            return {"Not Sure/Others": 0}
+            count += 1
+        if count == 2:
+            return {"Not sure/Others": 100}
     return output
 
 image = gr.inputs.Image(shape=(224, 224))