Hev832 commited on
Commit
d2d174a
1 Parent(s): 6170420

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/pretrained_v2', 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
 
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