Update app.py

app.py

@@ -30,7 +30,7 @@ def predict(input_dict):
     prediction = model.predict(image)
     probs = tf.nn.softmax(prediction[0]).numpy()
 
-    if probs[0] < 0.5:
+    if probs.max() < 0.5:
         return "Please try again."
     else:
         return {str(i): float(probs[i]) for i in range(10)}