Update app.py

app.py

@@ -459,9 +459,9 @@ def create_chatbot_demo():
                 output_vis = gr.HighlightedText(
                     label="Denoising Process Visualization",
                     combine_adjacent=False, # Keep tokens separate
-                    show_legend=False,      # Hide default legend table
-                    color_map=color_map,    # Provide the color map
-                    elem_classes="token-hl" # Add class for token styling
+                    show_legend=True,      # Hide default legend table
+                    #color_map=color_map,    # Provide the color map
+                    #elem_classes="token-hl" # Add class for token styling
                 )
                 # Use Markdown to display the custom legend
                 gr.Markdown(legend_html)