ialhashim commited on
Commit
56346a3
·
verified ·
1 Parent(s): 6de8932

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from skimage import io, color
10
  import cv2
11
  from colorizer import normalize_lab_channels, torch_normalized_lab_to_rgb
12
 
13
- model = torch.load('colorizer.pth')
14
  model = model.eval()
15
 
16
  def colorize_img( img_float32, model, res=512, border=0.2, apply_blur=True):
 
10
  import cv2
11
  from colorizer import normalize_lab_channels, torch_normalized_lab_to_rgb
12
 
13
+ model = torch.load('colorizer.pth', weights_only=False)
14
  model = model.eval()
15
 
16
  def colorize_img( img_float32, model, res=512, border=0.2, apply_blur=True):