asigalov61 commited on
Commit
601785e
·
verified ·
1 Parent(s): b0d55e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -366,26 +366,26 @@ with gr.Blocks() as demo:
366
 
367
  #==================================================================================
368
 
369
- gr.Markdown("## Upload seed MIDI or click 'Generate' button for random output")
370
 
371
  input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
372
 
373
- gr.Markdown("## Generate")
374
 
375
- num_gen_tokens = gr.Slider(15, 1024, value=1024, step=1, label="Number of tokens to generate")
376
  model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
377
 
378
  generate_btn = gr.Button("Generate", variant="primary")
379
 
380
- gr.Markdown("## Results")
381
 
382
- output_audio = gr.Audio(label="Output MIDI audio", format="wav", elem_id="midi_audio")
383
- output_plot = gr.Plot(label="Output MIDI score plot")
384
- output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
385
 
386
  generate_btn.click(Generate_Accompaniment,
387
  [input_midi,
388
- num_gen_tokens,
389
  model_temperature
390
  ],
391
  [
 
366
 
367
  #==================================================================================
368
 
369
+ gr.Markdown("## Upload source melody MIDI")
370
 
371
  input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
372
 
373
+ gr.Markdown("## Generation options")
374
 
375
+ generation_type = gr.Radio(["Guided", "Freestyle"] value="Guided", step=1, label="Generation type")
376
  model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
377
 
378
  generate_btn = gr.Button("Generate", variant="primary")
379
 
380
+ gr.Markdown("## Generation results")
381
 
382
+ output_audio = gr.Audio(label="MIDI audio", format="wav", elem_id="midi_audio")
383
+ output_plot = gr.Plot(label="MIDI score plot")
384
+ output_midi = gr.File(label="MIDI file", file_types=[".mid"])
385
 
386
  generate_btn.click(Generate_Accompaniment,
387
  [input_midi,
388
+ generation_type,
389
  model_temperature
390
  ],
391
  [