Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from omegaconf import OmegaConf
 from transformers import pipeline
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
@@ -22,6 +23,7 @@ OmegaConf.register_new_resolver("load_pipe", load_pipe)
 
 models_config = OmegaConf.to_object(OmegaConf.load("configs/models.yaml"))
 
+@spaces.GPU
 def automatic_speech_recognition(model_id: str, dialect_id: str, audio_file: str):
     model = models_config[model_id]["model"]
     generate_kwargs = {