Update app.py

app.py

@@ -6,8 +6,11 @@ pipe = pipeline("text-to-speech", model="suno/bark-small")
 
 def text_to_speech(text):
     output = pipe(text)
-    audio = Audio(output[0]["audio"], rate=output[0]["sampling_rate"])
-    return audio
+    if len(output) > 0 and "audio" in output[0] and "sampling_rate" in output[0]:
+        audio = Audio(output[0]["audio"], rate=output[0]["sampling_rate"])
+        return audio
+    else:
+        return None
 
 iface = gr.Interface(
     fn=text_to_speech,