Update FantasyTalking/infer.py

FantasyTalking/infer.py

@@ -164,7 +164,6 @@ def load_models(args):
     return pipe, fantasytalking, wav2vec_processor, wav2vec
 
 
-
 def main(args, pipe, fantasytalking, wav2vec_processor, wav2vec):
     os.makedirs(args.output_dir, exist_ok=True)
 
@@ -183,12 +182,16 @@ def main(args, pipe, fantasytalking, wav2vec_processor, wav2vec):
     )
     audio_proj_split, audio_context_lens = fantasytalking.split_tensor_with_padding(
         audio_proj_fea, pos_idx_ranges, expand_length=4
-    )  # [b,21,9+8,768]
+    )
 
-    # Image-to-video
+    if pipe is None:
+        print("❌ Pipe is None: Video generation skipped because Wan model is not loaded.")
+        return None
+
+    # إذا كان عندك pipe نكمل توليد الفيديو
     video_audio = pipe(
         prompt=args.prompt,
-        negative_prompt="人物静止不动，静止，色调艳丽，过曝，静态，细节模糊不清，字幕，风格，作品，画作，画面，静止，整体发灰，最差质量，低质量，JPEG压缩残留，丑陋的，残缺的，多余的手指，画得不好的手部，画得不好的脸部，畸形的，毁容的，形态畸形的肢体，手指融合，静止不动的画面，杂乱的背景，三条腿，背景人很多，倒着走",
+        negative_prompt="...",
         input_image=image,
         width=width,
         height=height,
@@ -227,6 +230,7 @@ def main(args, pipe, fantasytalking, wav2vec_processor, wav2vec):
     return save_path
 
 
+
 if __name__ == "__main__":
     args = parse_args()
     pipe, fantasytalking, wav2vec_processor, wav2vec = load_models(args)