Fix dataframe headers

app.py

@@ -47,7 +47,7 @@ def analysis(Text):
 MODEL_MAPPING = {
     'Random': RandomAnalyser(),
     'RoBERTa': RoBERTaAnalyser(),
-    'ChatGPT': ChatGPTAnalyser(),
+    'ChatGPT': RandomAnalyser(),
 }
 
 OUR_MODEL = 'RoBERTa'
@@ -108,7 +108,7 @@ with gr.Blocks(title=TITLE) as demo:
             cm_plot = gr.Plot(label='Confusion Matrix')
 
     with gr.Row():
-        dataframe = gr.Dataframe(type="pandas", wrap=True)
+        dataframe = gr.Dataframe(type="pandas", wrap=True, headers=['Text', 'Label', 'Predict'])
 
     input_n_samples.change(fn=classify, inputs=input_n_samples, outputs=[dataframe, bar_plot, cm_plot])