layout fix

generator.py

@@ -71,6 +71,7 @@ class Generator:
             "hovermode": "x",
 
         })
+        inp_len = len(input_ids[0])
         for i in range(max_length):
             is_caif_step = (
                 i % caif_period == 0 and self.caif_sampler is not None
@@ -97,7 +98,7 @@ class Generator:
                 scatter_data = go.Scatter({
                     "x": list(range(len(gen_history))),
                     "y": gen_history,
-                    "hovertext": [self.tokenizer.decode(t) for t in input_ids[0]]
+                    "hovertext": [self.tokenizer.decode(t) for t in input_ids[0][inp_len:]]
                 })
                 fig = go.Figure([scatter_data], layout=layout)
                 plot.plotly_chart(fig, use_container_width=True)