Update app.py

app.py

@@ -103,19 +103,23 @@ def ID_MIDI(input_midi):
     output_midi_src_dataset= 'Unknown'
     output_midi_path_str = 'None'
     
-    if new_midi_md5hash in MIDID_database:
+    if new_midi_md5hash in midid_md5_hashes:
 
-        client = InferenceClient(api_key=HF_TOKEN)
-        
-        prompt = "Please create a summary table for a MIDI file based on the following keywords strings, best possible description and best possible summary fields. Please respond with the table only. Do not say anything else. Thank you."
+        midid_entry_idx = midid_md5_hashes.index(new_midi_md5hash)
+
+        MIDID_record = midid_dataset[midid_entry_idx]
 
-        output_midi_records_count = len(MIDID_database[new_midi_md5hash])
+        output_midi_records_count = len(MIDID_record)
 
-        output_entry = random.choice(MIDID_database[new_midi_md5hash])
+        output_entry = random.choice(MIDID_record)
 
-        output_midi_src_dataset = output_entry['midi_dataset']
+        output_midi_src_dataset = output_entry[0]
 
-        output_midi_path_str = output_entry['midi_path']
+        output_midi_path_str = output_entry[1]
+
+        client = InferenceClient(api_key=HF_TOKEN)
+
+        prompt = "Please create a summary table for a MIDI file based on the following keywords strings, best possible description and best possible summary fields. Please respond with the table only. Do not say anything else. Thank you."
         
         data = 'Source MIDI dataset: ' + output_midi_src_dataset + '\n' + output_midi_path_str
         
@@ -176,9 +180,9 @@ if __name__ == "__main__":
     print('=' * 70)
 
     print('Loading MIDID database...')
-    midid_dataset = load_dataset("asigalov61/MIDID")
+    midid_dataset = load_dataset("asigalov61/MIDID")['train']
 
-    midid_md5_hashes = midid_dataset['train']['midi_hash']
+    midid_md5_hashes = midid_dataset['midi_hash']
 
     print('Done!')
     print('=' * 70)