Hugging Face's logo

Spaces:
dhairyashah
/
deepfake-detection-gpu
Sleeping

App Files Community
aaronespasa commited on
Commit
248f463
·
1 Parent(s): 02c225a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ model = InceptionResnetV1(
29
  device=DEVICE
30
  )
31
 
32
- checkpoint = torch.load("resnetinceptionv1_epoch_32.pth")
33
  model.load_state_dict(checkpoint['model_state_dict'])
34
  model.to(DEVICE)
35
  model.eval()
 
29
  device=DEVICE
30
  )
31
 
32
+ checkpoint = torch.load("resnetinceptionv1_epoch_32.pth", map_location=torch.device('cpu'))
33
  model.load_state_dict(checkpoint['model_state_dict'])
34
  model.to(DEVICE)
35
  model.eval()