Update app.py

app.py

@@ -73,13 +73,13 @@ def gen_sources(deepfake_img):
     #------------Initialize:Decoder-F------------------------
     checkpoint_path_f = "./models/model_vaq1_ff.pth"
     # Load model checkpoints
-    checkpoint_f = torch.load(checkpoint_path_f, map_location=self.device)
+    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.eval()
     #------------Initialize:Decoder-G------------------------
     checkpoint_path_g = "./models/model_vaq2_gg.pth"
-    checkpoint_g = torch.load(checkpoint_path_g, map_location=self.device)
+    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.eval()