fix

app.py

@@ -4,6 +4,8 @@ import argparse
 from datetime import datetime
 import librosa
 from infer import load_models,main
+import spaces
+
 try:
     import torch
     if torch.cuda.is_available():
@@ -12,7 +14,7 @@ except Exception as e:
     print(f"GPU warmup failed: {e}")
 
 pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
-
+@spaces.GPU
 def generate_video(
     image_path,
     audio_path,