Add GPU decorator to run_inference function and import spaces module

app.py

@@ -3,7 +3,7 @@ import os
 import subprocess
 import time
 from typing import Dict, List, Optional, Tuple
-
+import spaces
 import gradio as gr
 import soundfile as sf
 import torch
@@ -117,6 +117,7 @@ def _validate_condition_choices(condition_type: Optional[List[str]]) -> List[str
     return condition_type
 
 
+@spaces.GPU
 def run_inference(
     prompt_text: str,
     condition_audio: Optional[str],