Update app.py

app.py

@@ -4,7 +4,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 from snac import SNAC
 
-@spaces.GPU
 def redistribute_codes(row):
     """
     Convert a sequence of token codes into an audio waveform using SNAC.
@@ -62,6 +61,7 @@ models = {
     },
 }
 
+@spaces.GPU
 def generate_audio(text, temperature, top_p, max_new_tokens, model_name):
     """
     Given input text and model parameters, generate speech audio using the chosen model.