Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import shutil
 from huggingface_hub import snapshot_download
@@ -12,6 +13,7 @@ print(os.listdir(hallo_dir))
 
 from scripts.inference import predict
 
+@spaces.GPU(duration=120)
 def run_inference(source_image, driving_audio, progress=gr.Progress(track_tqdm=True)):
         
     return predict(source_image, driving_audio, 1.0, 1.0, 1.0, 1.2)