Update app.py

app.py

@@ -20,6 +20,8 @@ def generate(
             "role": "system",
             "content": system_input,
         },
+        {"role": "user", "content": "Hello there!"},
+        {"role": "assistant", "content": "Hi!"},
         {"role": "user", "content": user_input},
     ]
     prompt = pipe.tokenizer.apply_chat_template(message_template, tokenize=False, add_generation_prompt=True)
@@ -27,14 +29,14 @@ def generate(
                    temperature=temperature, top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty)
     return outputs[0]["generated_text"]
 
-model_choices = ["Felladrin/Pythia-31M-Chat-v1", "Felladrin/Llama-160M-Chat-v1", "Felladrin/Smol-Llama-101M-Chat-v1", "Felladrin/TinyMistral-248M-SFT-v4"]
+model_choices = ["Felladrin/Llama-160M-Chat-v1", "Felladrin/Smol-Llama-101M-Chat-v1", "Felladrin/TinyMistral-248M-SFT-v4", "Felladrin/Pythia-31M-Chat-v1"]
 
 g = gr.Interface(
     fn=generate,
     inputs=[
         gr.components.Dropdown(choices=model_choices, label="Model", value=model_choices[0], interactive=True),
-        gr.components.Textbox(lines=2, label="System Message", value="You are a highly knowledgeable and friendly chatbot equipped with extensive information across various domains. Your goal is to understand and respond to user inquiries with accuracy and clarity. You're adept at providing detailed explanations, concise summaries, and insightful responses. Your interactions are always respectful, helpful, and focused on delivering the most relevant information to the user."),
-        gr.components.Textbox(lines=2, label="User Message", value="How many planets are in our solar system?"),
+        gr.components.Textbox(lines=2, label="System Message", value="You are a highly knowledgeable and friendly assistant. Your goal is to understand and respond to user inquiries with accuracy and clarity. You're adept at providing detailed explanations, concise summaries, and insightful responses. Your interactions are always respectful, helpful, and focused on delivering the most relevant information to the user."),
+        gr.components.Textbox(lines=2, label="User Message", value="Tell me something curious about the Earth!"),
         gr.components.Slider(minimum=0, maximum=1, value=0.4, label="Temperature"),
         gr.components.Slider(minimum=0, maximum=1, value=0.25, label="Top p"),
         gr.components.Slider(minimum=0, maximum=100, step=1, value=7, label="Top k"),
@@ -42,7 +44,7 @@ g = gr.Interface(
         gr.components.Slider(minimum=1, maximum=1024, step=1, value=256, label="Max tokens"),
     ],
     outputs=[gr.Textbox(lines=10, label="Output")],
-    title="A space to try out models fine-tuned by Felladrin",
+    title="A place to try out text-generation models fine-tuned by Felladrin",
     concurrency_limit=1
 )