Update app.py

app.py

@@ -12,6 +12,7 @@ import os
 import copy
 import statistics
 import random
+import pickle
 
 import time as reqtime
 import datetime
@@ -51,10 +52,11 @@ SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 print('Loading MIDI Loops Small Dataset...')
 print('=' * 70)
 
-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'
-                                    )
+midi_loops_dataset = pickle.load(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!')