layout fix

generator.py

@@ -67,7 +67,9 @@ class Generator:
                 "title": "Desired Attribute"
             },
             "plot_bgcolor": '#FFFFFF',
-            "template": "plotly_white"
+            "template": "plotly_white",
+            "hovermode": "x",
+
         })
         for i in range(max_length):
             is_caif_step = (
@@ -94,7 +96,8 @@ class Generator:
                 gen_history += [probs]
                 scatter_data = go.Scatter({
                     "x": list(range(len(gen_history))),
-                    "y": gen_history
+                    "y": gen_history,
+                    "hovertext": [self.tokenizer.decode(t) for t in input_ids[0]]
                 })
                 fig = go.Figure([scatter_data], layout=layout)
                 plot.plotly_chart(fig, use_container_width=True)