Update app.py

app.py

@@ -75,13 +75,13 @@ def gen_sources(deepfake_img):
     # Load model checkpoints
     checkpoint_f = torch.load(checkpoint_path_f, map_location=device)
     # Load the state dictionary into the models
-    model_vaq_f = model_vaq_f.load_state_dict(checkpoint_f, strict=True)
+    model_vaq_f.load_state_dict(checkpoint_f, strict=True)
     model_vaq_f.eval()
     #------------Initialize:Decoder-G------------------------
     checkpoint_path_g = "./models/model_vaq2_gg.pth"
     checkpoint_g = torch.load(checkpoint_path_g, map_location=device)
     # Load the state dictionary into the models
-    model_vaq_g = model_vaq_g.load_state_dict(checkpoint_g, strict=True)
+    model_vaq_g.load_state_dict(checkpoint_g, strict=True)
     model_vaq_g.eval()
     ##------------------------Initialize Model-F-------------------------------------
     model_z1 = DeepfakeToSourceTransformer().to(device)