Underscore only sole space patches

app.py

@@ -321,11 +321,12 @@ class BytelatentProcessor:
 
             # Create highlighted text data
             _highlighted_data, patch_count = self._create_highlight_data(patch_lengths, tokens)
-            ind_highlighted_data = [(text.replace(" -1", "_-1").replace("-1", ""), label) for text, label in _highlighted_data]
+            ind_highlighted_data = [(text.replace("-1", ""), label) for text, label in _highlighted_data]
             grouped_data = defaultdict(str)
             for text, label in ind_highlighted_data:
                 grouped_data[label] += text
-            highlighted_data = [(text, label) for label, text in grouped_data.items()]
+            underscore_sole_space_patch = lambda text: text.replace(" ", "_") if text == " " else text
+            highlighted_data = [(underscore_sole_space_patch(text), label) for label, text in grouped_data.items()]
 
             # Create plot
             fig = None