Spaces:
Sleeping
Sleeping
fix background tasks
Browse files
server.py
CHANGED
|
@@ -5,7 +5,7 @@ import random
|
|
| 5 |
from fastapi import FastAPI
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from fastapi.responses import FileResponse
|
| 8 |
-
from fastapi
|
| 9 |
from starlette.requests import Request
|
| 10 |
from kokoro import KPipeline
|
| 11 |
import soundfile as sf
|
|
@@ -80,10 +80,13 @@ def health_check():
|
|
| 80 |
async def receive(input_load: InputLoad, request: Request) -> FileResponse:
|
| 81 |
audio_path = get_audio(input_load.text, input_load.language)
|
| 82 |
|
|
|
|
|
|
|
|
|
|
| 83 |
return FileResponse(
|
| 84 |
path=audio_path,
|
| 85 |
media_type="audio/wav",
|
| 86 |
filename="generated_audio.wav",
|
| 87 |
-
background=
|
| 88 |
)
|
| 89 |
|
|
|
|
| 5 |
from fastapi import FastAPI
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from fastapi.responses import FileResponse
|
| 8 |
+
from fastapi import BackgroundTasks
|
| 9 |
from starlette.requests import Request
|
| 10 |
from kokoro import KPipeline
|
| 11 |
import soundfile as sf
|
|
|
|
| 80 |
async def receive(input_load: InputLoad, request: Request) -> FileResponse:
|
| 81 |
audio_path = get_audio(input_load.text, input_load.language)
|
| 82 |
|
| 83 |
+
background_tasks = BackgroundTasks()
|
| 84 |
+
background_tasks.add_task(cleanup_temp_file, audio_path)
|
| 85 |
+
|
| 86 |
return FileResponse(
|
| 87 |
path=audio_path,
|
| 88 |
media_type="audio/wav",
|
| 89 |
filename="generated_audio.wav",
|
| 90 |
+
background=background_tasks
|
| 91 |
)
|
| 92 |
|