asigalov61 commited on
Commit
8b5450d
·
verified ·
1 Parent(s): 6f4c5be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -59,7 +59,7 @@ print('=' * 70)
59
  MODEL_CHECKPOINTS = {
60
  'with velocity - 3 epochs': 'Monster_Piano_Transformer_Velocity_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth',
61
  'without velocity - 3 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_50647_steps_0.8166_loss_0.7561_acc.pth',
62
- 'without velocity - 5 epochs': ''
63
  }
64
 
65
  SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -523,7 +523,8 @@ with gr.Blocks() as demo:
523
 
524
  gr.Markdown("## Generate")
525
 
526
- model_selector = gr.Dropdown(["without velocity - 3 epochs",
 
527
  "with velocity - 3 epochs"
528
  ],
529
  label="Select model",
 
59
  MODEL_CHECKPOINTS = {
60
  'with velocity - 3 epochs': 'Monster_Piano_Transformer_Velocity_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth',
61
  'without velocity - 3 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_50647_steps_0.8166_loss_0.7561_acc.pth',
62
+ 'without velocity - 5 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_84419_steps_0.7474_loss_0.7782_acc.pth'
63
  }
64
 
65
  SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
523
 
524
  gr.Markdown("## Generate")
525
 
526
+ model_selector = gr.Dropdown(["without velocity - 5 epochs",
527
+ "without velocity - 3 epochs",
528
  "with velocity - 3 epochs"
529
  ],
530
  label="Select model",