Update app.py

app.py

@@ -143,6 +143,14 @@ async def generate(request: Request, body: GenerateRequest):
             audio_bytes = audio_bytesio.getvalue()
             return Response(content=audio_bytes, media_type="audio/wav")
 
+        elif body.task_type == "text-to-audio":
+            generator = pipeline("text-to-audio", model=model, tokenizer=tokenizer, device=device)
+            audio = generator(body.input_text)
+            audio_bytesio = BytesIO()
+            sf.write(audio_bytesio, audio["sampling_rate"], np.int16(audio["audio"]))
+            audio_bytes = audio_bytesio.getvalue()
+            return Response(content=audio_bytes, media_type="audio/wav")
+        
         elif body.task_type == "text-to-video":
             try:
                 generator = pipeline("text-to-video", model=model, tokenizer=tokenizer, device=device)