Post process debug

app.py

@@ -24,15 +24,19 @@ def postprocess(prediction):
 
 def predict_fn(image):
     image = preprocess(image)
-    prediction = model(image)
+    prediction = model(image)[0].numpy()
     print('****************')
     print(prediction)
+    try:
+        print({labels[i]: prediction[i] for i in range(len(labels))})
+    except:
+        print("default gives error")
     print('****************')
-    print(prediction[0])
-    print('****************')
-    print(prediction[0].numpy())
-    print('****************')
-    print(prediction.numpy())
+    print(list(prediction))
+    try:
+        print({labels[i]: list(prediction)[i] for i in range(len(labels))})
+    except:
+        print("list gives error")
     scores = postprocess(prediction)
     return scores