Update app.py

app.py

@@ -30,7 +30,8 @@ from modules.denormalize import denormalize_bin, denormalize_tr, denormalize_ar
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-config = OmegaConf.load(self.config_path)
+config_path = "./models/config.yaml"
+config = OmegaConf.load(config_path)
 # Extract parameters specific to GumbelVQ
 vq_params = config.model.params
 # Initialize the GumbelVQ models
@@ -70,7 +71,6 @@ def gen_sources(deepfake_img):
     # Convert segmented numpy array (BGR) back to PIL Image
     deepfake_seg = Image.fromarray(cv2.cvtColor(segmented_np, cv2.COLOR_BGR2RGB))
     #------------Initialize:Decoder-F------------------------
-    config_path = "./models/config.yaml"
     checkpoint_path_f = "./models/model_vaq1_ff.pth"
     # Load model checkpoints
     checkpoint_f = torch.load(self.checkpoint_path_f, map_location=self.device)