Update scripts/inference.py

scripts/inference.py

@@ -9,7 +9,7 @@ This script can be run from the command line with the following command:
 
 python scripts/app.py
 """
-
+import spaces
 import gradio as gr
 import argparse
 import copy
@@ -405,7 +405,7 @@ def inference(cfg, image_processor, audio_processor, pipeline, audioproj, save_d
     tensor_to_video(tensor_result, output_file, audio_path)
     return output_file
 
-
+@spaces.GPU()
 def get_model(cfg: argparse.Namespace) -> None:
     """
     Trains the model using the given configuration (cfg).