Hugging Face's logo

Spaces:
not-lain
/
deepfake-detection
Running

App Files Community
aaronespasa commited on
Commit
48359f9
•
1 Parent(s): 4f7c4d9

Update app.py

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