mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 21:45:00 +03:00
22 lines
603 B
Python
Executable File
22 lines
603 B
Python
Executable File
|
|
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)
|
|
|