Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import numpy as np
 import torch
@@ -35,7 +36,8 @@ def synthesise(text):
     inputs = processor(text=text, return_tensors="pt")
     speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
     return speech.cpu()
-    
+
+@spaces.GPU
 def speech_to_speech_translation(audio):
   # print(model)
   translated_text = translate(model, audio)