Update app.py

app.py

@@ -86,8 +86,8 @@ def generate_charts(ner_output_bin: dict) -> Tuple[go.Figure, np.ndarray]:
     bin_sizes = list(entity_counts_bin.values())
 
     bin_color_map = {
-      "External": "#00008B", // Deep blue
-      "Internal": "#8B0000"  // Deep red
+      "External": "#00008B",
+      "Internal": "#8B0000"
     }
 
     bin_colors = [bin_color_map.get(label, "#FFFFFF") for label in bin_labels]
@@ -182,9 +182,10 @@ iface = gr.Interface(
     inputs=gr.Textbox(lines=5, label="Input Text", placeholder="Write about how your breakfast went or anything else that happened or might happen to you ..."),
     outputs=[
         gr.HighlightedText(label="Binary Sequence Classification",
-                           color_map={
-                               "External": "#6ad5bcff",
-                               "Internal": "#ee8bacff"}
+                           bin_color_map = {
+                              "External": "#00008B",
+                              "Internal": "#8B0000"
+                            }
                           ),
         gr.Plot(label="Binary SeqClass Entity Count Bar Chart"),
         gr.Image(label="Entity Word Cloud")