Add GPU decorator to run_pipeline and update_metrics methods in GradioInterface

interface.py

@@ -2,6 +2,7 @@ import time
 import uuid
 
 import gradio as gr
+import spaces
 import yaml
 
 from characters import CHARACTERS
@@ -212,6 +213,7 @@ class GradioInterface:
         self.current_voice = self.svs_model_map[self.current_svs_model]["voices"][voice]
         return gr.update(value=voice)
 
+    @spaces.GPU
     def run_pipeline(self, audio_path):
         if not audio_path:
             return gr.update(value=None), gr.update(value=None)
@@ -228,6 +230,7 @@ class GradioInterface:
             value=self.results["output_audio_path"]
         )
 
+    @spaces.GPU
     def update_metrics(self, audio_path):
         if not audio_path or not self.results:
             return gr.update(value="")