Update app.py

app.py

@@ -245,10 +245,7 @@ def process_input(text_input, file):
 
     df = pd.DataFrame(results)
 
-    # Áp dụng định dạng hiển thị với 3 chữ số phần thập phân cho cột 'Xác suất'
-    styler = df.style.format({'Xác suất': '{:.3f}'})
-
-    return styler 
+    return df 
 
 with gr.Blocks() as demo:
     gr.Markdown("## Phân tích cảm xúc")
@@ -257,7 +254,7 @@ with gr.Blocks() as demo:
     gr.Markdown("Hoặc tải lên tệp .txt hoặc .csv chứa các bình luận:")
     file_input = gr.File(label="Tải tệp", file_types=[".txt", ".csv"])
     predict_button = gr.Button("Dự đoán")
-    output_table = gr.Dataframe(interactive=False)
+    output_table = gr.Dataframe(headers=["Comment", "Dự đoán", "Xác suất"], interactive=False)
 
     predict_button.click(
         fn=process_input,