Fixing postprocess

app.py

@@ -19,8 +19,8 @@ def preprocess(image):
     return image / 255.
 
 def postprocess(prediction):
-    # return {labels[i]: prediction[i] for i in range(len(labels))}
-    return {labels[i]: 0 for i in range(len(labels))}
+    return {labels[i]: prediction[i] for i in range(len(labels))}
+    # return {labels[i]: 0 for i in range(len(labels))}
 
 def predict_fn(image):
     image = preprocess(image)
@@ -28,12 +28,14 @@ def predict_fn(image):
     print('****************')
     print(prediction)
     try:
+        print("default")
         print({labels[i]: prediction[i] for i in range(len(labels))})
     except:
         print("default gives error")
     print('****************')
     print(list(prediction))
     try:
+        print("list")
         print({labels[i]: list(prediction)[i] for i in range(len(labels))})
     except:
         print("list gives error")