fix

app.py

@@ -4,7 +4,12 @@ import argparse
 from datetime import datetime
 import librosa
 from infer import load_models,main
-
+try:
+    import torch
+    if torch.cuda.is_available():
+        _ = torch.tensor([0.0]).to('cuda')
+except Exception as e:
+    print(f"GPU warmup failed: {e}")
 
 pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None