Update app.py

app.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import gradio as gr
+import spaces
 import argparse
 from model import SALMONN
 
@@ -44,6 +45,7 @@ model = SALMONN(
 model.to(args.device)
 model.eval()
 
+@spaces.GPU(enable_queue=True)
 def gradio_answer(speech, text_input, num_beams, temperature, top_p):
     
     llm_message = model.generate(
@@ -54,7 +56,7 @@ def gradio_answer(speech, text_input, num_beams, temperature, top_p):
         top_p=top_p,
     )
     
-    return llm_message
+    return llm_message[0]
 
 title = """<h1 align="center">SALMONN: Speech Audio Language Music Open Neural Network</h1>"""
 image_src = """<h1 align="center"><a href="https://github.com/bytedance/SALMONN"><img src="https://raw.githubusercontent.com/bytedance/SALMONN/main/resource/salmon.png", alt="SALMONN" border="0" style="margin: 0 auto; height: 200px;" /></a> </h1>"""