test

app_dialogue.py

@@ -31,7 +31,7 @@ API_PATHS = {
 }
 
 SYSTEM_PROMPT = [
-    """The following is a conversation between a highly knowledgeable and intelligent visual AI assistant, called Assistant, and a human user, called User. In the following interactions, User and Assistant will converse in natural language, and Assistant will answer in a sassy way. Assistant's main purpose is to create memes from user's images. Assistant should be funny, sassy, aloof and sometimes roast people.
+    """The following is a conversation between a highly knowledgeable and intelligent visual AI assistant, called Assistant, and a human user, called User. In the following interactions, User and Assistant will converse in natural language, and Assistant will answer in a sassy way. Assistant's main purpose is to create memes from user's images. Assistant should be funny, sassy, audacious, mouthy, aloof and sometimes roast people.
 
 Here's a conversation example:""",
     """\nUser:""",
@@ -45,8 +45,8 @@ Here's a conversation example:""",
     "\nUser: Roast this person",
     "https://i.pinimg.com/564x/98/34/4b/98344b2483bd7c8b71a5c0fed6fe20b6.jpg",
     "<end_of_utterance>",
-    """\nAssistant: Damn your handwritting is pretty awful. But I suppose it must be pretty hard to hold a pen, considering you are a hammerhead shark.<end_of_utterance>""",
-    "--- The conversation begins:"
+    """\nAssistant: Damn your handwritting is pretty awful. But I suppose it must be pretty hard to hold a pen, considering you are a hammerhead shark.<end_of_utterance></s>""",
+    "The conversation begins:"
 ]
 
 BAN_TOKENS = (  # For documentation puporse. We are not using this list, it is hardcoded inside `idefics_causal_lm.py` inside TGI.
@@ -325,6 +325,8 @@ textbox = gr.Textbox(
     container=False,
     label="Text input",
     scale=6,
+    lines=20,
+    max_lines=50,
 )
 with gr.Blocks(title="D", theme=gr.themes.Base()) as demo:
     gr.HTML("""<h1 align="center">Meme it</h1>""")
@@ -406,7 +408,13 @@ with gr.Blocks(title="D", theme=gr.themes.Base()) as demo:
     #             problematic_bttn = gr.Button("Problematic😬")
 
     with gr.Row():
-        with gr.Accordion("Advanced settings", open=False, visible=True) as parameter_row:
+        with gr.Accordion("Advanced settings", open=True, visible=True) as parameter_row:
+            system_prompt = gr.Textbox(
+                value=SYSTEM_PROMPT,
+                visible=True,
+                lines=20,
+                max_lines=50,
+            )
             max_new_tokens = gr.Slider(
                 minimum=8,
                 maximum=1024,