Spaces:

molher
/

TorchFakes

Sleeping

molher commited on Jun 4, 2022

Commit

878334f

•

1 Parent(s): 052af59

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import matplotlib.pyplot as plt
 DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
 print(f'Running on device: {DEVICE.upper()}')
-torch.load('./models/resnetinceptionv1_final.pth',map_location='cpu')
 mtcnn = MTCNN(
     select_largest=False,

 DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
 print(f'Running on device: {DEVICE.upper()}')
+torch.load('resnetinceptionv1_final.pth',map_location='cpu')
 mtcnn = MTCNN(
     select_largest=False,