Update app.py

app.py

@@ -221,8 +221,8 @@ def model_infer(img_name):
         image, mask = input
 
         image_vis = image.transpose(1, 2, 0)
-        gt_mask = mask[0]
-        #gt_mask = mask[0].astype('uint8')
+        #gt_mask = mask[0]
+        gt_mask = mask[0].astype('uint8')
         pr_mask = (output[0].numpy() > threshold).astype('uint8')[0]
         i = i + 1
         if i >= break_at: