Update app.py

app.py

@@ -60,7 +60,8 @@ def predict_class(image):
 def classify_image(image):
     results = predict_class(image)
     output = {labels.get(i): float(results[i]) for i in range(len(results))}
-    return output if max(output.values()) >=1 else {"NO_CIFAR10_CLASS": 1}
+    result = output if max(output.values()) >=0.98 else {"NO_CIFAR10_CLASS": 1}
+    return result
 
 
 inputs = gr.inputs.Image(type="pil", label="Upload an image")