init

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 import torchaudio
 from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
-
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_name = "ibm-granite/granite-speech-3.3-8b"
@@ -23,7 +23,7 @@ def _load_audio_mono_16k(file_path: str) -> torch.Tensor:
         wav = torchaudio.functional.resample(wav, sr, 16000)
     return wav
 
-
+@spaces.GPU
 def process_audio(audio_path: str, instruction: str, max_tokens: int = 200) -> str:
     if not audio_path:
         return "Please upload an audio file."