Update app.py

app.py

@@ -97,7 +97,8 @@ if __name__ == "__main__":
                     label="Highlighted generation",
                     combine_adjacent=True,
                     show_legend=True,
-                ).style(color_map=label_to_color)
+                    color_map=label_to_color,
+                )
 
         button.click(get_tokens_and_labels, inputs=prompt, outputs=highlighted_text)