voice-changer/demo/mods/Trainer_Model.py

14 lines
380 B
Python
Raw Normal View History

2022-11-08 13:00:47 +03:00
from fastapi.responses import FileResponse
import os
def mod_get_model(modelFile:str):
2022-12-09 06:52:27 +03:00
modelPath = os.path.join("MMVC_Trainer/logs", modelFile)
2022-11-08 13:00:47 +03:00
return FileResponse(path=modelPath)
def mod_delete_model(modelFile:str):
2022-12-09 06:52:27 +03:00
modelPath = os.path.join("MMVC_Trainer/logs", modelFile)
2022-11-08 13:00:47 +03:00
os.unlink(modelPath)
return {"Model deleted": f"{modelFile}"}