Update difpoint/inference.py

difpoint/inference.py

@@ -162,19 +162,19 @@ class Inferencer(object):
         self.device = 'cuda'
         from difpoint.model import get_model
         self.point_diffusion = get_model()
-        ckpt = torch.load('/home/yinuo/Gradio-UI_copy/difpoint/outputs/2024.08.26_dim_70_frame_64_vox1_selected_d6.5_c8.5/2024-08-26--16-52-34/checkpoint-500000.pth')
+        ckpt = torch.load('./downloaded_repo/ckpts/KDTalker.pth', weights_only=False)
 
         self.point_diffusion.load_state_dict(ckpt['model'])
         print('model', self.point_diffusion.children())
         self.point_diffusion.eval()
         self.point_diffusion.to(self.device)
 
-        lm_croper_checkpoint = os.path.join('difpoint/dataset_process/ckpts/', 'shape_predictor_68_face_landmarks.dat')
+        lm_croper_checkpoint = './downloaded_repo/ckpts/shape_predictor_68_face_landmarks.dat'
         self.croper = Croper(lm_croper_checkpoint)
 
         self.norm_info = dict(np.load(r'difpoint/datasets/norm_info_d6.5_c8.5_vox1_train.npz'))
 
-        wav2lip_checkpoint = 'difpoint/dataset_process/ckpts/wav2lip.pth'
+        wav2lip_checkpoint = './downloaded_repo/ckpts/wav2lip.pth'
         self.wav2lip_model = AudioEncoder(wav2lip_checkpoint, 'cuda')
         self.wav2lip_model.cuda()
         self.wav2lip_model.eval()