Update app.py

app.py

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