Spaces:

molher
/

TorchFakes

Sleeping

molher commited on Jun 4, 2022

Commit

e0006c0

1 Parent(s): 878334f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ model = InceptionResnetV1(
     num_classes=1,
     device=DEVICE
 )
-model.load_state_dict(torch.load('./models/resnetinceptionv1_final.pth',map_location='cpu'))
 model.to(DEVICE)
 model.eval()
 print("MTCNN & Classfier models loaded")

     num_classes=1,
     device=DEVICE
 )
+model.load_state_dict(torch.load('resnetinceptionv1_final.pth',map_location='cpu'))
 model.to(DEVICE)
 model.eval()
 print("MTCNN & Classfier models loaded")