Hugging Face's logo

Spaces:
Chakshu123
/
image-colorization-with-hint
Running

App Files Community
Chakshu123 commited on
Commit
d40e27d
•
1 Parent(s): 8c65b27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -143,7 +143,7 @@ class ECCVGenerator(BaseColor):
143
 
144
  # model_net = torch.load(f'weights/colorizer.pt')
145
  model_net = ECCVGenerator()
146
- model_net.load_state_dict(torch.load(f'weights/colorizer (1).pt', map_loaction=torch.device('cpu')))
147
 
148
 
149
  def resize_original(img: Image.Image):
 
143
 
144
  # model_net = torch.load(f'weights/colorizer.pt')
145
  model_net = ECCVGenerator()
146
+ model_net.load_state_dict(torch.load(f'weights/colorizer (1).pt', map_location=torch.device('cpu')))
147
 
148
 
149
  def resize_original(img: Image.Image):