Update app.py

app.py

@@ -45,7 +45,7 @@ def predict_2(image):
             
         image = Image.fromarray(image).resize((size,size))
         image = np.array(image)        
-        stride = 2
+        stride = 1
         steps = int(pach_size/stride)
         patches_img = patchify(image, (pach_size, pach_size, 3), step=steps)  #Step=256 for 256 patches means no overlap        
         patches_img = patches_img[:,:,0,:,:,:]
@@ -77,7 +77,7 @@ def predict_2(image):
       
         unpatched_prediction = unpatchify(patched_prediction, (image.shape[0], image.shape[1]))
         labels = LABEL_NAMES
-        res = list(map(str, np.bincount(unpatched_prediction.reshape(-1))/unpatched_prediction.reshape(-1).shape[0]))
+        res = np.bincount(unpatched_prediction.reshape(-1))/unpatched_prediction.reshape(-1).shape[0]
         out = dict(list(zip(labels, res)))
         unpatched_prediction = targets_classes_colors[unpatched_prediction]