add markdown

app.py

@@ -61,17 +61,26 @@ if __name__ == "__main__":
     with demo:
         gr.Markdown(
             """
-            # Foo Bar
+            # 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 🤗
+
+            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.
             """
         )
 
-        prompt = gr.Textbox(label="Prompt", lines=3, value="Today is")
-        highlighted_text = gr.HighlightedText(
-            label="Highlighted generation",
-            combine_adjacent=True,
-            show_legend=True,
-        ).style(color_map=label_to_color)
-        button = gr.Button(f"Generate with {MODEL_NAME}")
+        with gr.Row():
+            with gr.Column():
+                prompt = gr.Textbox(label="Prompt", lines=3, value="Today is")
+                button = gr.Button(f"Generate with {MODEL_NAME}")
+            with gr.Column():
+                highlighted_text = gr.HighlightedText(
+                    label="Highlighted generation",
+                    combine_adjacent=True,
+                    show_legend=True,
+                ).style(color_map=label_to_color)
 
         button.click(get_tokens_and_labels, inputs=prompt, outputs=highlighted_text)