Update app.py

app.py

@@ -45,6 +45,10 @@ def generate_first(text, voice='af_heart', speed=1, use_gpu=CUDA_AVAILABLE):
         return (24000, audio.numpy()), ps
     return None, ''
 
+# Arena API
+def predict(text, voice='af_heart', speed=1):
+    return generate_first(text, voice, speed, use_gpu=False)[0]
+
 def tokenize_first(text, voice='af_heart'):
     words = text.split()  # This splits the text into words based on spaces
     return words  # Return a list of words