Show correct labels

utils.py

@@ -47,6 +47,7 @@ scheduler = CommitScheduler(
 )
 
 df = pd.read_csv(os.path.join(LOCAL_DIR,'data.csv'))
+
 filenames = list(os.path.join(LOCAL_DIR, 'songs') + '/' + df.filename + '.mp3')
 
 indices = list(df.index)
@@ -181,7 +182,7 @@ def fake_last_response(
     vote_last_response(
       state, "fake", request
     )
-    return (disable_btn,) * 3 + (gr.Markdown(f"### Fake \nModel : {state.row.algorithm}", visible=True),)
+    return (disable_btn,) * 3 + (gr.Markdown(f"### {state.row.label} \nModel : {state.row.algorithm}", visible=True),)
 
 def real_last_response(
     state, request: gr.Request
@@ -189,5 +190,5 @@ def real_last_response(
     vote_last_response(
       state, "real", request
     )
-    return (disable_btn,) * 3 + (gr.Markdown(f"### Real", visible=True),)
+    return (disable_btn,) * 3 + (gr.Markdown(f"### {state.row.label}", visible=True),)