Update app.py

app.py

@@ -6,7 +6,7 @@ model_name = "google/flan-t5-large"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-def concatenate_and_generate(source_text, target_example_texts, reranking, temperature, top_p):
+def run_tinystyler(source_text, target_example_texts, reranking, temperature, top_p):
     concatenated_text = source_text + " " + target_example_texts
     inputs = tokenizer(concatenated_text, return_tensors="pt")
     
@@ -23,8 +23,8 @@ def concatenate_and_generate(source_text, target_example_texts, reranking, tempe
     return generated_text
 
 # Preset examples with cached generations
-preset_examples = [
-    {
+preset_examples = {
+    "Example 1": {
         "source_text": "Once upon a time in a small village",
         "target_example_texts": "In a land far away, there was a kingdom ruled by a wise king. Every day, the people of the kingdom would gather to listen to the king's stories, which were full of wisdom and kindness.",
         "reranking": 5,
@@ -32,7 +32,7 @@ preset_examples = [
         "top_p": 1.0,
         "output": "Once upon a time in a small village in a land far away, there was a kingdom ruled by a wise king. Every day, the people of the kingdom would gather to listen to the king's stories, which were full of wisdom and kindness."
     },
-    {
+    "Example 2": {
         "source_text": "The quick brown fox",
         "target_example_texts": "A nimble, chocolate-colored fox swiftly darted through the emerald forest, weaving between trees with grace and agility.",
         "reranking": 5,
@@ -40,13 +40,16 @@ preset_examples = [
         "top_p": 0.9,
         "output": "The quick brown fox, a nimble, chocolate-colored fox, swiftly darted through the emerald forest, weaving between trees with grace and agility."
     }
-]
+}
 
 # Define Gradio interface
 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 results, at slower speed.")
-
+    
+    with gr.Row():
+        example_dropdown = gr.Dropdown(label="Examples", choices=list(preset_examples.keys()))
+    
     source_text = gr.Textbox(lines=3, placeholder="Enter the source text to transform into the target style...", label="Source Text")
     target_example_texts = gr.Textbox(lines=5, placeholder="Enter example texts of the target style (one per line)...", label="Example Texts of the Target Style")
     reranking = gr.Slider(1, 10, value=5, step=1, label="Re-ranking")
@@ -55,18 +58,17 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
     
     output = gr.Textbox(lines=5, placeholder="Click 'Generate' to transform the source text into the target style.", label="Output", interactive=False)
 
-    def set_example(example):
+    def set_example(example_name):
+        example = preset_examples[example_name]
         return example["source_text"], example["target_example_texts"], example["reranking"], example["temperature"], example["top_p"], example["output"]
 
-    example_dropdown = gr.Dropdown(label="Preset Examples", choices=[f"Example {i+1}" for i in range(len(preset_examples))])
-    example_button = gr.Button("Load Example")
-    example_button.click(
-        lambda example_index: set_example(preset_examples[int(example_index.split()[-1])-1]),
+    example_dropdown.change(
+        set_example,
         inputs=[example_dropdown],
         outputs=[source_text, target_example_texts, reranking, temperature, top_p, output]
     )
     
     btn = gr.Button("Generate")
-    btn.click(concatenate_and_generate, [source_text, target_example_texts, reranking, temperature, top_p], output)
+    btn.click(run_tinystyler, [source_text, target_example_texts, reranking, temperature, top_p], output)
 
 demo.launch()