Update app.py

app.py

@@ -1,5 +1,6 @@
 from snac import SNAC
 import torch
+import spaces
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import snapshot_download
@@ -62,6 +63,7 @@ def process_prompt(prompt, voice, tokenizer, device):
     return modified_input_ids.to(device), attention_mask.to(device)
 
 # Generate speech tokens
+spaces.GPU()
 def generate_speech_tokens(input_ids, attention_mask, model, params):
     with torch.no_grad():
         generated_ids = model.generate(