chore: add descriptions

app.py

@@ -10,6 +10,14 @@ from threading import Thread
 import gradio as gr
 import time
 
+MARKDOWN_DESCRIPTION = """\
+# SDPrompt-RetNet-v2-beta
+
+This is a demo of [SDPrompt-RetNet-v2-beta](https://huggingface.co/isek-ai/SDPrompt-RetNet-v2-beta), a pretrained RetNet model trained on [danbooru tags](isek-ai/danbooru-tags-2016-2023) dataset.
+
+This model can only complete tags after the input text, so you have to start with a tag like `1girl` or `1boy, 1girl`. Also, this model generates tags in alphabetical order, you should place tags in alphabetical order in the input text.
+"""
+
 MODEL_NAME = "isek-ai/SDPrompt-RetNet-v2-beta"
 DEVICE = "cpu"
 
@@ -17,7 +25,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.bfloat16,
-    device_map={"": 0},
+    device_map=DEVICE,
     trust_remote_code=True,
 )
 model.config.use_cache = True
@@ -107,6 +115,8 @@ COPY_ACTION_JS = """\
 
 def demo():
     with gr.Blocks() as ui:
+        gr.Markdown(MARKDOWN_DESCRIPTION)
+
         with gr.Column():
             input_text = gr.Textbox(label="Input", value="1girl")
 
@@ -133,9 +143,9 @@ def demo():
                     max_new_length_slider = gr.Slider(
                         label="Max new length",
                         minimum=4,
-                        maximum=384,
+                        maximum=512,
                         step=4,
-                        value=128,
+                        value=160,
                     )
                     do_sample_checkbox = gr.Checkbox(
                         label="Do sample",
@@ -177,6 +187,17 @@ def demo():
                         value=6,
                     )
 
+        gr.Examples(
+            [
+                "1girl, blue hair",
+                "1girl, animal ears",
+                "1girl, detached sleeves",
+                "1boy, 1girl",
+                "1other",
+            ],
+            inputs=[input_text],
+        )
+
         btn.click(
             fn=generate,
             inputs=[