pujanpaudel commited on
Commit
0f466f0
·
verified ·
1 Parent(s): 9e182b5

added new model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ class DeepFakeModel(L.LightningModule):
51
 
52
  # Load trained model
53
  model = DeepFakeModel(hyper_params)
54
- state_dict = torch.load("model.pth", map_location=torch.device(device))
55
  model.load_state_dict(state_dict)
56
  model.to(device)
57
  model.eval()
 
51
 
52
  # Load trained model
53
  model = DeepFakeModel(hyper_params)
54
+ state_dict = torch.load("deepfake_new_trained.pth", map_location=torch.device(device))
55
  model.load_state_dict(state_dict)
56
  model.to(device)
57
  model.eval()