Update app.py

app.py

@@ -114,7 +114,7 @@ def main_func(inputs):
         
         final_mask = masks[0]
         mask_colors = np.zeros((final_mask.shape[0], final_mask.shape[1], 3), dtype=np.uint8)
-        mask_colors[final_mask, :] = np.array([[128, 0, 0]])
+        mask_colors[final_mask, :] = np.array([[256, 0, 0]])
         return Image.fromarray((mask_colors+ image_input).astype('uint8'), 'RGB')
     else:
         print('Prediction:: No vehicle found in the image')