Update app.py

app.py

@@ -49,7 +49,7 @@ class DeepFakeModel(L.LightningModule):
 
 # Load trained model
 model = DeepFakeModel(hyper_params)
-state_dict = torch.load("trained_model.pth", map_location=torch.device(device))
+state_dict = torch.load("deepfake_new_trained.pth", map_location=torch.device(device))
 model.load_state_dict(state_dict)
 model.to(device)
 model.eval()