Update app.py

app.py

@@ -30,10 +30,12 @@ def process(input_path):
 
 
     res = results[1]["response"]["tagging"]["2"]
-    
+
+    dict_labels = {}
     moods = []
     for m in res['moods']:
       moods.append((m['name'], m['probability']))
+      dict_labels[m['name']] = m['probability']
     print(moods)
     
     music_desc=[]
@@ -68,7 +70,7 @@ def process(input_path):
     audioquality = [res['audio_quality'][0]['name'], res['audio_quality'][0]['probability']]
     print(audioquality)
 
-    return res['moods']
+    return dict_labels
 
 
 demo = gr.Interface(fn=process,