Hugging Face's logo

Spaces:
IanNathaniel
/
Zero-DCE
Runtime error

App Files Community
IanNathaniel commited on
Commit
683e48b
•
1 Parent(s): a844295

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def lowlight(image_path):
31
  data_lowlight = data_lowlight.cpu().unsqueeze(0)
32
 
33
  DCE_net = model.enhance_net_nopool().cpu()
34
- DCE_net.load_state_dict(torch.load('Epoch99.pth'))
35
  start = time.time()
36
  _,enhanced_image,_ = DCE_net(data_lowlight)
37
 
 
31
  data_lowlight = data_lowlight.cpu().unsqueeze(0)
32
 
33
  DCE_net = model.enhance_net_nopool().cpu()
34
+ DCE_net.load_state_dict(torch.load('Epoch99.pth', map_location=torch.device('cpu')))
35
  start = time.time()
36
  _,enhanced_image,_ = DCE_net(data_lowlight)
37