🌈

app.py

@@ -61,10 +61,11 @@ if __name__ == "__main__":
     with demo:
         gr.Markdown(
             """
-            # Color Coded Text Generation
+            # 🌈 Color Coded Text Generation 🌈
 
-            This is a demo of how you can obtain the probabilities of each token being generated, and use them to
-            color code the generated text 🟢🟡🔴. Feel free to clone this demo and modify it to your needs 🤗
+            This is a demo of how you can obtain the probabilities of each generated token, and use them to
+            color code the model output.
+            Feel free to clone this demo and modify it to your needs 🤗
 
             Internally, it relies on [`compute_transition_scores`](https://huggingface.co/docs/transformers/main/en/main_classes/text_generation#transformers.GenerationMixin.compute_transition_scores),
             which was added in `transformers` v4.26.0.