More interface changes with updated labels

app.py

@@ -45,13 +45,13 @@ audio_examples = [
 ]
 
 df_init = pd.DataFrame(columns=['Task', 'Top 1', 'Top 2', 'Top 3', 'Top 4', 'Top 5'])
-transcription_df = gr.DataFrame(value=df_init, label="Output Dataframe", row_count=(
+transcription_df = gr.DataFrame(value=df_init, label="Model Results", row_count=(
     0, "dynamic"), max_rows=30, wrap=True, overflow_row_behaviour='paginate')
 # outputs = [gr.components.Textbox()]
 outputs = transcription_df
 
 df_init_live = pd.DataFrame(columns=['Task', 'Top 1', 'Top 2', 'Top 3', 'Top 4', 'Top 5'])
-transcription_df_live = gr.DataFrame(value=df_init_live, label="Output Dataframe", row_count=(
+transcription_df_live = gr.DataFrame(value=df_init_live, label="Model Results", row_count=(
     0, "dynamic"), max_rows=30, wrap=True, overflow_row_behaviour='paginate')
 outputs_live = transcription_df_live
 
@@ -100,7 +100,15 @@ ID2CLASS = {
 
 }
 
-TASKS = ['GS', 'MTGInstrument', 'MTGGenre', 'MTGTop50', 'MTGMood', 'NSynthI', 'NSynthP', 'VocalSetS', 'VocalSetT','EMO',]
+#TASKS = ['GS', 'MTGInstrument', 'MTGGenre', 'MTGTop50', 'MTGMood', 'NSynthI', 'NSynthP', 'VocalSetS', 'VocalSetT','EMO',]
+TASKS = ['GS', 'MTGInstrument', 'MTGGenre', 'MTGMood', 'EMO']
+TASK_LABELS = {
+    'GS': 'Scale',
+    'MTGInstrument': 'Instruments',
+    'MTGGenre': 'Genre',
+    'MTGMood': 'Mood',
+    'EMO','Emotion (Valence/Arousal'
+}
 Regression_TASKS = ['EMO']
 head_dir = './Prediction_Head/best-layer-MERT-v1-95M'
 for task in TASKS:
@@ -159,7 +167,7 @@ def model_infernce(inputs):
         # task_output_texts = task_output_texts + f"TASK {task} output:\n" + "\n".join([str(ID2CLASS[task][str(sorted_idx[idx].item())])+f', probability: {sorted_prob[idx].item():.2%}' for idx in range(top_n_show)]) + '\n'
         # task_output_texts = task_output_texts + '----------------------\n'
 
-        row_elements = [task]
+        row_elements = [TASK_LABELS[task]]
         for idx in range(top_n_show):
             print(ID2CLASS[task])
             # print('id', str(sorted_idx[idx].item()))