Update app.py

app.py

@@ -16,6 +16,18 @@ pipe = pipeline(
     device=device,
 )
 
+
+id2label = {
+    "LABEL_0" : 'drumbass', 
+    "LABEL_1" : 'dubtechno', 
+    "LABEL_2": 'dupstep', 
+    "LABEL_3":'hardcore_breaks', 
+    "LABEL_4": 'house', 
+    "LABEL_5":'psytrance', 
+    "LABEL_6" : 'techno', 
+    "LABEL_7":'ukgarage'
+}
+
 def get_edm(filepath):
     output = pipe(
         filepath,
@@ -23,7 +35,7 @@ def get_edm(filepath):
         chunk_length_s=30,
         batch_size=8,
     )
-    return output[0]["label"]
+    return id2label[output[0]["label"]]
 
 
 demo = gr.Blocks()