SebastianSchramm commited on
Commit
bfd2ec1
·
unverified ·
1 Parent(s): 6efa3b5

fix background tasks

Browse files
Files changed (1) hide show
  1. server.py +5 -2
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.background import BackgroundTasks
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=BackgroundTasks(cleanup_temp_file, audio_path)
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