Update app.py

app.py

@@ -21,7 +21,7 @@ os.environ['USE_FLASH_ATTENTION'] = '1'
 
 import torch
 
-torch.set_float32_matmul_precision('high')
+torch.set_float32_matmul_precision('medium')
 torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
 torch.backends.cudnn.allow_tf32 = True # allow tf32 on cudnn
 torch.backends.cuda.enable_mem_efficient_sdp(True)
@@ -108,7 +108,10 @@ def load_midi(input_midi):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi.name)
     
-    escore_notes = TMIDIX.advanced_score_processor(raw_score, return_enhanced_score_notes=True)
+    escore_notes = TMIDIX.advanced_score_processor(raw_score, 
+                                                   return_enhanced_score_notes=True, 
+                                                   apply_sustain=True
+                                                  )
     
     escore_notes = TMIDIX.augment_enhanced_score_notes(escore_notes[0], timings_divider=16)
 
@@ -534,6 +537,11 @@ def reset(final_composition=[], generated_batches=[], block_lines=[]):
     
 #==================================================================================
 
+def clear():
+    return None, None, None, [], []
+
+#==================================================================================
+
 def reset_demo(final_composition=[], generated_batches=[], block_lines=[]):
     
     final_composition = []
@@ -629,6 +637,7 @@ with gr.Blocks() as demo:
     
     add_btn = gr.Button("Add batch", variant="primary")
     remove_btn = gr.Button("Remove batch", variant="stop")
+    clear_btn = gr.ClearButton()
     
     final_audio_output = gr.Audio(label="Final MIDI audio", format="mp3", elem_id="midi_audio")
     final_plot_output = gr.Plot(label="Final MIDI plot")
@@ -639,7 +648,12 @@ with gr.Blocks() as demo:
 
     remove_btn.click(remove_batch, [batch_number, num_gen_tokens, final_composition, generated_batches, block_lines], 
                      [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines])
-
+    
+    clear_btn.click(clear, 
+                    inputs=None, 
+                    outputs=[final_audio_output, final_plot_output, final_file_output, final_composition, block_lines]
+                   )
+    
     demo.unload(reset_demo)
 
 #==================================================================================