mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-24 22:15:02 +03:00
22 lines
605 B
Python
22 lines
605 B
Python
|
|
||
|
|
from fastapi.responses import JSONResponse
|
||
|
|
from fastapi.encoders import jsonable_encoder
|
||
|
|
from trainer_mods.files import get_file_list
|
||
|
|
import os
|
||
|
|
|
||
|
|
def mod_get_models():
|
||
|
|
gModels = get_file_list(f'/MMVC_Trainer/logs/G*.pth')
|
||
|
|
dModels = get_file_list(f'/MMVC_Trainer/logs/D*.pth')
|
||
|
|
models = []
|
||
|
|
models.extend(gModels)
|
||
|
|
models.extend(dModels)
|
||
|
|
models = [ os.path.basename(x) for x in models]
|
||
|
|
|
||
|
|
models = sorted(models)
|
||
|
|
data = {
|
||
|
|
"models":models
|
||
|
|
}
|
||
|
|
json_compatible_item_data = jsonable_encoder(data)
|
||
|
|
return JSONResponse(content=json_compatible_item_data)
|
||
|
|
|