feat: more interesting examples

app.py

@@ -17,7 +17,9 @@ import gradio as gr
 MODEL_NAME = os.environ.get("MODEL_NAME", None)
 assert MODEL_NAME is not None
 MODEL_PATH = hf_hub_download(repo_id=MODEL_NAME, filename="model.safetensors")
-DEVICE = torch.device("cuda")
+DEVICE = (
+    torch.device("mps") if torch.backends.mps.is_available() else torch.device("cuda")
+)
 
 BAD_WORD_KEYWORDS = ["(medium)"]
 
@@ -64,6 +66,7 @@ def demo():
         text: str,
         auto_detect: bool,
         copyright_tags: str = "",
+        length: str = "short",
         max_new_tokens: int = 128,
         do_sample: bool = False,
         temperature: float = 0.1,
@@ -72,7 +75,7 @@ def demo():
     ):
         tag_text = (
             "<|bos|>"
-            "<|aspect_ratio:tall|><|rating:general|><|length:long|>"
+            f"<|aspect_ratio:tall|><|rating:general|><|length:{length}|>"
             "<|reserved_2|><|reserved_3|><|reserved_4|>"
             "<|translate:exact|><|input_end|>"
             "<copyright>" + copyright_tags.strip()
@@ -112,22 +115,36 @@ def demo():
 
     # warmup
     print("warming up...")
-    print(generate_tags("Miku is looking at viewer.", True))
+    print(generate_tags("Hatsune Miku is looking at viewer.", True))
     print("done.")
 
     with gr.Blocks() as ui:
         with gr.Column():
             with gr.Row():
                 with gr.Column():
-                    text = gr.Text(label="Text", lines=4)
+                    text = gr.Text(
+                        label="Text",
+                        lines=4,
+                        placeholder="A girl with fox ears and tail in maid costume is looking at viewer.",
+                    )
                     auto_detect = gr.Checkbox(
                         label="Auto detect copyright tags.", value=False
                     )
                     copyright_tags = gr.Textbox(
                         label="Copyright tags",
-                        placeholder="Enter copyright tags here. e.g.) hatsune miku",
+                        placeholder="You can specify copyright tags manually. e.g.) hatsune miku",
+                    )
+                    length = gr.Dropdown(
+                        label="Length",
+                        choices=[
+                            "very_short",
+                            "short",
+                            "long",
+                            "very_long",
+                        ],
+                        value="short",
                     )
-                    translate_btn = gr.Button(value="Translate")
+                    translate_btn = gr.Button(value="Translate", variant="primary")
 
                     with gr.Accordion(label="Advanced", open=False):
                         max_new_tokens = gr.Number(label="Max new tokens", value=128)
@@ -160,37 +177,60 @@ def demo():
 
             gr.Examples(
                 examples=[
-                    ["Miku is looking at viewer.", True, ""],
+                    [
+                        "A girl with fox ears and tail in maid costume is looking at viewer.",
+                        False,
+                        "",
+                        "short",
+                    ],
+                    [
+                        "Hatsune Miku is looking at viewer.",
+                        True,
+                        "",
+                        "short",
+                    ],
                     [
                         "Fujita Kotone, Tsukimura Temari, Hanami Saki from Gakuen Idolmaster. They are in the hole, there are some tables and chairs. One's face is shaded, one is crying, and one is 😊.",
                         True,
                         "",
+                        "short",
                     ],
                     [
                         "A single girl wearing red hood is sleeping in the forest. View angle from above. grass field. many colorful flowers. Bright atmosphere.",
                         False,
                         "",
+                        "short",
+                    ],
+                    [
+                        "Arona and Plana are hugging each other.",
+                        True,
+                        "blue archive",
+                        "short",
+                    ],
+                    [
+                        "Arona and Plana are hugging each other.",
+                        True,
+                        "blue archive",
+                        "very_long",
                     ],
-                    ["Arona and Plana are hugging each other.", True, "blue archive"],
                     [
                         "There are two girls. A vivacious blonde gyaru leans against a classroom desk, her flashy accessories jingling as she gestures animatedly. Across from her stands the prim and proper class representative, her long black hair neatly framing her face as she listens attentively, occasionally adjusting her glasses with a delicate touch.",
                         False,
                         "",
+                        "long",
                     ],
                     [
-                        "1girl, solo, white and blue medium hair with side braid, dark blue parka with hoodie, looking at somewhere else viewer, cowboy shot",
+                        "1girl, solo, white and blue medium hair with side braid, dark blue parka with hoodie, looking at somewhere else viewer, cowboy shot, with many cats. the girl has cat ears and tail.",
                         False,
-                        ""
-                    ]
+                        "",
+                        "short",
+                    ],
                 ],
-                inputs=[text, auto_detect, copyright_tags],
+                inputs=[text, auto_detect, copyright_tags, length],
             )
 
         gr.on(
             triggers=[
-                # text.change,
-                # auto_detect.change,
-                # copyright_tags.change,
                 translate_btn.click,
             ],
             fn=generate_tags,
@@ -198,6 +238,7 @@ def demo():
                 text,
                 auto_detect,
                 copyright_tags,
+                length,
                 max_new_tokens,
                 do_sample,
                 temperature,