Update app.py

app.py

@@ -28,6 +28,8 @@ import TMIDIX
 
 from midi_to_colab_audio import midi_to_colab_audio
 
+from huggingface_hub import hf_hub_download
+
 import gradio as gr
 
 print('=' * 70)
@@ -49,7 +51,10 @@ SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 print('Loading MIDI Loops Small Dataset...')
 print('=' * 70)
 
-midi_loops_dataset = TMIDIX.Tegridy_Any_Pickle_File_Reader('MIDI-Loops-Dataset-Small-CC-BY-NC-SA.pickle')
+midi_loops_dataset = hf_hub_download(repo_id='asigalov61/MIDI-Loops', 
+                                     filename='MIDI_Loops_Processed_Dataset_116909_Loops_CC_BY_NC_SA.pickle',
+                                     repo_type='dataset'
+                                    )
 
 print('=' * 70)
 print('Done!')
@@ -145,15 +150,15 @@ def Mix_Loops(max_num_loops,
     
     all_chords_chunks = []
     
-    midi_loops = [l for l in midi_loops_dataset if len(set(l[1][0])) >= chunk_chords_set]
+    midi_loops = [l for l in midi_loops_dataset if len(set(l[1])) >= chunk_chords_set]
     
     for loop in tqdm.tqdm(midi_loops):
+        
         fn = loop[0]
         chords = loop[1]
         score = loop[2]
-    
-        for c in chords:
-            all_chords_chunks.append(c[:chunk_len])
+        
+        all_chords_chunks.append(chords[:chunk_len])
     
     all_chords_chunks = np.array(all_chords_chunks)
 
@@ -454,7 +459,7 @@ with gr.Blocks() as demo:
     max_num_loops = gr.Slider(2, 10, value=4, step=1, label="Maximum number of loops to mix")
     comp_loops_mult = gr.Slider(2, 4, value=2, step=1, label="Number of loops repetitions")
     chords_chunks_len = gr.Slider(4, 8, value=5, step=1, label="Number of loops chords to match")
-    loops_chords_set_len = gr.Slider(10, 20, value=13, step=1, label="Minimum number of unique chords in each loop")
+    loops_chords_set_len = gr.Slider(16, 28, value=13, step=1, label="Minimum number of unique chords in each loop")
     
     mix_btn = gr.Button("Mix", variant="primary")