Update app.py
Browse files
app.py
CHANGED
|
@@ -26,6 +26,7 @@ DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
|
|
| 26 |
mtcnn = MTCNN(select_largest=False, post_process=False, device=DEVICE).to(DEVICE).eval()
|
| 27 |
|
| 28 |
model = InceptionResnetV1(pretrained="vggface2", classify=True, num_classes=1, device=DEVICE)
|
|
|
|
| 29 |
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth", map_location=torch.device('cpu'))
|
| 30 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 31 |
model.to(DEVICE)
|
|
|
|
| 26 |
mtcnn = MTCNN(select_largest=False, post_process=False, device=DEVICE).to(DEVICE).eval()
|
| 27 |
|
| 28 |
model = InceptionResnetV1(pretrained="vggface2", classify=True, num_classes=1, device=DEVICE)
|
| 29 |
+
# Model Credits: https://huggingface.co/spaces/dhairyashah/deepfake-alpha-version/blob/main/CREDITS.md
|
| 30 |
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth", map_location=torch.device('cpu'))
|
| 31 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 32 |
model.to(DEVICE)
|