Fixing prob type

app.py

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