Update app.py
Browse files
app.py
CHANGED
@@ -467,7 +467,7 @@ with gr.Blocks(theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",ne
|
|
467 |
interactive=True,
|
468 |
)
|
469 |
with gr.Accordion(label="Change pretrains", open=False):
|
470 |
-
pretrained = lambda sr, letter: [os.path.abspath(os.path.join('assets/
|
471 |
pretrained_G14 = gr.Dropdown(
|
472 |
label="pretrained G",
|
473 |
# Get a list of all pretrained G model files in assets/pretrained_v2 that end with .pth
|
|
|
467 |
interactive=True,
|
468 |
)
|
469 |
with gr.Accordion(label="Change pretrains", open=False):
|
470 |
+
pretrained = lambda sr, letter: [os.path.abspath(os.path.join('assets/pretrained', file)) for file in os.listdir('assets/pretrained_v2') if file.endswith('.pth') and sr in file and letter in file]
|
471 |
pretrained_G14 = gr.Dropdown(
|
472 |
label="pretrained G",
|
473 |
# Get a list of all pretrained G model files in assets/pretrained_v2 that end with .pth
|