Update difpoint/inference.py

difpoint/inference.py

@@ -169,8 +169,8 @@ class Inferencer(object):
         self.point_diffusion.eval()
         self.point_diffusion.to(self.device)
 
-        lm_croper_checkpoint = './downloaded_repo/ckpts/shape_predictor_68_face_landmarks.dat'
-        self.croper = Croper(lm_croper_checkpoint)
+        self.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'))
 
@@ -274,6 +274,7 @@ class Inferencer(object):
     # 2024.06.26
     @torch.no_grad()
     def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t,  save_path='results'):
+        self.croper = Croper(self.lm_croper_checkpoint)
         self.live_portrait_pipeline.init()
         if audio_type == 'upload':
             audio_path = upload_audio_path