Update app.py

app.py

@@ -59,7 +59,7 @@ def predict_class(image):
 
 # UI Design for multy class output
 def classify_image(image):
-    results = predict_class(image)
+    results = predict_class(image)[0]
     print(results)
     output = {labels.get(i): float(results[i]) for i in range(len(results))}
     result = output if max(output.values()) >=0.98 else {"NO_CIFAR10_CLASS": 1}