Update app.py

app.py

@@ -108,7 +108,7 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
 
     gr.Markdown("# TinyStyler Demo")
     gr.Markdown(
-        "Style transfer the source text into the target style, given some example texts of the target style. You can adjust re-ranking and top_p to your desire to control the quality of style transfer. A higher re-ranking value will generally result in better generations, at slower speed.\n\n*Please note: this demo runs on a CPU-only machine, generation is much faster when run locally with a GPU.*"
+        "Style transfer the source text into the target style, given some example texts of the target style. You can adjust re-ranking and top_p to your desire to control the quality of style transfer. A higher re-ranking value will generally result in better generations, at slower speed.\n\nYou can find the model at [https://huggingface.co/tinystyler/tinystyler](https://huggingface.co/tinystyler/tinystyler).\n\n*Please note: this demo runs on a CPU-only machine, generation is much faster when run locally with a GPU.*"
     )
 
     with gr.Row():
@@ -161,6 +161,10 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
         output,
     )
 
+    # Pre-warm all examples
+    for preset_example in preset_examples.values():
+        run_tinystyler("A test.", preset_example["target_texts"], 1, 1.0, 1.0)
+
     # Initialize the fields with the first example
     (
         example_dropdown.value,