Update app.py

app.py

@@ -72,7 +72,7 @@ def pred_maps(images):
         if device == 'cuda':
             pred_tensor = pred_tensor.cpu()
         pred_tensor = torch.nn.functional.interpolate(pred_tensor.unsqueeze(0), size=image_shape, mode='bilinear', align_corners=True).squeeze().numpy()
-        pred_tensor = cv2.cvtColor((pred_tensor*255).astype(np.uint8)
+        pred_tensor = (pred_tensor*255).astype(np.uint8)
         cv2.imwrite(save_path, pred_tensor)
 
     zip_file_path = os.path.join(save_dir, "{}.zip".format(save_dir))