2022-11-08 13:00:47 +03:00
|
|
|
|
|
|
|
|
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():
|
2022-12-09 06:52:27 +03:00
|
|
|
gModels = get_file_list(f'MMVC_Trainer/logs/G*.pth')
|
|
|
|
|
dModels = get_file_list(f'MMVC_Trainer/logs/D*.pth')
|
2022-11-08 13:00:47 +03:00
|
|
|
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)
|
|
|
|
|
|
