chore: add examples

app.py

@@ -7,7 +7,7 @@ from transformers import (
     AutoModelForPreTraining,
     AutoProcessor,
     AutoConfig,
-    PreTrainedTokenizerFast
+    PreTrainedTokenizerFast,
 )
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
@@ -21,15 +21,20 @@ DEVICE = torch.device("cuda")
 
 BAD_WORD_KEYWORDS = ["(medium)"]
 
+
 def fix_compiled_state_dict(state_dict: dict):
     return {k.replace("._orig_mod.", "."): v for k, v in state_dict.items()}
 
+
 def get_bad_words_ids(tokenizer: PreTrainedTokenizerFast):
     ids = [
-        [id] for token, id in tokenizer.vocab.items() if any(word in token for word in BAD_WORD_KEYWORDS)
+        [id]
+        for token, id in tokenizer.vocab.items()
+        if any(word in token for word in BAD_WORD_KEYWORDS)
     ]
     return ids
 
+
 def prepare_models():
     config = AutoConfig.from_pretrained(MODEL_NAME, trust_remote_code=True)
     model = AutoModelForPreTraining.from_config(
@@ -89,6 +94,7 @@ def demo():
             temperature=temperature,
             top_k=top_k,
             top_p=top_p,
+            no_repeat_ngram_size=1,
             eos_token_id=processor.decoder_tokenizer.eos_token_id,
             pad_token_id=processor.decoder_tokenizer.pad_token_id,
             bad_words_ids=ban_ids,
@@ -153,8 +159,31 @@ def demo():
                     time_elapsed = gr.Markdown(value="")
 
             gr.Examples(
-                examples=[["Miku is looking at viewer.", True]],
-                inputs=[text, auto_detect],
+                examples=[
+                    ["Miku is looking at viewer.", True, ""],
+                    [
+                        "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,
+                        "",
+                    ],
+                    [
+                        "A single girl wearing red hood is sleeping in the forest. View angle from above. grass field. many colorful flowers. Bright atmosphere.",
+                        False,
+                        "",
+                    ],
+                    ["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,
+                        "",
+                    ],
+                    [
+                        "1girl, solo, white and blue medium hair with side braid, dark blue parka with hoodie, looking at somewhere else viewer, cowboy shot",
+                        False,
+                        ""
+                    ]
+                ],
+                inputs=[text, auto_detect, copyright_tags],
             )
 
         gr.on(