Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import soundfile as sf
 import numpy as np
@@ -6,6 +7,7 @@ from dia.model import Dia
 # Load the model once
 model = Dia.from_pretrained("nari-labs/Dia-1.6B")
 
+@spaces.GPU
 def generate_dialogue(script):
     output = model.generate(script)
     filename = "generated_dialogue.wav"