fix

generator.py

@@ -59,6 +59,15 @@ class Generator:
         text = st.empty()
         plot = st.empty()
         gen_history = []
+        layout = go.Layout({
+            "xaxis": {
+                "title": "# Tokens"
+            },
+            "yaxis": {
+                "title": "Desired Attribute"
+            },
+            "plot_bgcolor": '#FFFFFF',
+        })
         for i in range(max_length):
             is_caif_step = (
                 i % caif_period == 0 and self.caif_sampler is not None
@@ -86,6 +95,7 @@ class Generator:
                     "x": list(range(len(gen_history))),
                     "y": gen_history
                 })
+                fig = go.Figure([scatter_data], layout=layout)
                 plot.plotly_chart([scatter_data], use_container_width=True)
             text.text(current_decoded)