Update app.py

app.py

@@ -59,8 +59,10 @@ def predict_class(image):
 
 # UI Design for multy class output
 def classify_image(image):
-    results = predict_class(image)[0]
+    results = predict_class(image)
+    print("results is ...", results)
     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]