Update server.py

server.py

@@ -5,13 +5,11 @@ import torch
 import os
 from pipeline import build_audiosep, separate_audio
 
-app = FastAPI()
+# Ensure directories exist
+os.makedirs('uploaded', exist_ok=True)
+os.makedirs('processed', exist_ok=True)
 
-@app.on_event("startup")
-async def startup_event():
-    # Create directories for uploaded and processed files
-    os.makedirs('uploaded', exist_ok=True)
-    os.makedirs('processed', exist_ok=True)
+app = FastAPI()
 
 # Mount the static files directory
 app.mount("/processed", StaticFiles(directory="processed"), name="processed")
@@ -41,4 +39,4 @@ async def upload_file(file: UploadFile = File(...), text: str = Form(...)):
 
     # Response with processed audio file URL
     processed_file_url = f"/processed/{os.path.basename(processed_audio_file_path)}"
-    return JSONResponse(status_code=200, content={"message": "Audio file received and processed.", "processed_audio_file": processed_file_url})
+    return JSONResponse(status_code=200, content={"message": "Audio file received and processed.", "processed_audio_file": processed_file_url})