Update app.py

app.py

@@ -60,8 +60,7 @@ 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))}
-    result = output if max(output.values) >=1 else {"other": 1}
-    return output
+    return output if max(output.values()) >=1 else {"NO_CIFAR10_CLASS": 1}
 
 
 inputs = gr.inputs.Image(type="pil", label="Upload an image")