fix

generator.py

@@ -54,7 +54,7 @@ class Generator:
             input_prompt,
             num_samples,
         )
-
+        text = st.text()
         for i in range(max_length):
             is_caif_step = (
                 i % caif_period == 0 and self.caif_sampler is not None
@@ -70,7 +70,7 @@ class Generator:
             progress_bar.progress((i+1)/max_length)
             if ended_sequences.all():
                 break
-            st.write(self.tokenizer.decode(input_ids[0]))
+            text.text(self.tokenizer.decode(input_ids[0]))
 
         return (
             [