Update app.py

app.py

@@ -64,8 +64,8 @@ class AnimateController:
         # gate_dict_path = "ckpts/landscape.pt"    
         # self.pnp.set_audio_projector(gate_dict_path, audio_projector_path)   
 
-        self.audio_projector_path = "ckpts/audio_projector_landscape.pth"
-        self.adapter_ckpt_path = "ckpts/landscape.pt"        
+        # self.audio_projector_path = "ckpts/audio_projector_landscape.pth"
+        # self.adapter_ckpt_path = "ckpts/landscape.pt"        
         
         
     @spaces.GPU        
@@ -95,6 +95,10 @@ class AnimateController:
                  cfg_scale=5, image_path=None,
                  pnp_f_t=0.8, pnp_attn_t=0.8,):
 
+        if self.audio_projector_path is None:
+            self.audio_projector_path = "ckpts/audio_projector_landscape.pth"
+            self.adapter_ckpt_path = "ckpts/landscape.pt"               
+        
         print(f"before run_pnp {self.audio_projector_path} -- {self.adapter_ckpt_path}")        
             
         image = self.pnp.run_pnp(