Update app.py

app.py

@@ -91,7 +91,7 @@ midi_gas_pse_pickle = hf_hub_download(repo_id='asigalov61/MIDI-GAS',
                                       repo_type='dataset'
                                      )
 
-midi_gas_pse = [a[3] for a in TMIDIX.Tegridy_Any_Pickle_File_Reader(midi_gas_pse_pickle)]
+midi_gas_pse = np.array([a[3] for a in TMIDIX.Tegridy_Any_Pickle_File_Reader(midi_gas_pse_pickle)])
 
 print('=' * 70)
 print('Done!')
@@ -102,7 +102,7 @@ print('=' * 70)
 print('=' * 70)
 print('Instantiating model...')
 
-device_type = 'cpu'
+device_type = 'cuda'
 dtype = 'bfloat16'
 
 ptdtype = {'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
@@ -247,7 +247,7 @@ def select_best_output(outputs, embeddings, src_embeddings, top_k=10):
 
 #==================================================================================
 
-# @spaces.GPU
+@spaces.GPU
 def Classify_MIDI_Genre(input_midi,
                         input_melody,
                         melody_patch,
@@ -327,10 +327,10 @@ def Classify_MIDI_Genre(input_midi,
     print('Rendering results...')
     print('=' * 70)
 
-    sname = ' --- '.join(all_scores[result[1][0]][:3])
-    print('Sample entry', sname)
+    song_name = ' --- '.join(midi_gas_ps[result[1][0]][:3])
+    print('Song entry', song_name)
     
-    song = all_scores[result[1][0]][3]
+    song = midi_gas_ps[result[1][0]][3]
     print('Sample INTs', song[:15])
     print('=' * 70)
 
@@ -402,18 +402,18 @@ def Classify_MIDI_Genre(input_midi,
             
     
     audio = midi_to_colab_audio(new_fn, 
-                        soundfont_path=SOUDFONT_PATH,
-                        sample_rate=16000,
-                        volume_scale=10,
-                        output_for_gradio=True
-                        )
+                                soundfont_path=SOUDFONT_PATH,
+                                sample_rate=16000,
+                                volume_scale=10,
+                                output_for_gradio=True
+                                )
     
     print('Done!')
     print('=' * 70)
 
     #===============================================================================
 
-    output_title = str(sname)
+    output_title = str(song_name)
     output_midi = str(new_fn)
     output_audio = (16000, audio)