Fix system prompt input bug into generate(), and add translations of interface

app.py

@@ -12,9 +12,11 @@ DEFAULT_MAX_NEW_TOKENS = 512
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 
 DESCRIPTION = """\
-# Jacaranda / Swahili AI Assistant
+# Msaidizi wa AI ya Kiswahili
 
-This Space demonstrates the [Swahili (Jacaranda) model](https://huggingface.co/abhinand/tamil-llama-7b-instruct-v0.1) fine-tuned from Llama-2 7b, used as a daily life AI assistant.
+Hii inaonyesha kielelezo cha Kiswahili (Jacaranda) kilichoundwa kutoka Llama-2 7b, kinachotumiwa kama msaidizi wa AI kwa maisha ya kila siku.
+
+(This Space demonstrates the [Swahili (Jacaranda) model](https://huggingface.co/abhinand/tamil-llama-7b-instruct-v0.1) fine-tuned from Llama-2 7b, used as a daily life AI assistant.)
 """
 
 LICENSE = """
@@ -44,7 +46,6 @@ if torch.cuda.is_available():
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
-    system_prompt: str = SYSTEM_PROMPT,
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
     top_p: float = 0.9,
@@ -52,9 +53,7 @@ def generate(
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]:
     print("chat history: ", chat_history)
-    conversation = []
-    if system_prompt:
-        conversation.append({"role": "system", "content": system_prompt})
+    conversation = [{"role": "system", "content": SYSTEM_PROMPT}]
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
     conversation.append({"role": "user", "content": message})
@@ -94,62 +93,12 @@ examples = [
     ["Nivae nini kwenye harusi?"],
 ]
 
-# chat_interface = gr.ChatInterface(
-#     fn=generate,
-#     additional_inputs=[
-#         gr.Textbox(label="System prompt", lines=6),
-#         gr.Slider(
-#             label="Max new tokens",
-#             minimum=1,
-#             maximum=MAX_MAX_NEW_TOKENS,
-#             step=1,
-#             value=DEFAULT_MAX_NEW_TOKENS,
-#         ),
-#         gr.Slider(
-#             label="Temperature",
-#             minimum=0.1,
-#             maximum=4.0,
-#             step=0.1,
-#             value=0.6,
-#         ),
-#         gr.Slider(
-#             label="Top-p (nucleus sampling)",
-#             minimum=0.05,
-#             maximum=1.0,
-#             step=0.05,
-#             value=0.9,
-#         ),
-#         gr.Slider(
-#             label="Top-k",
-#             minimum=1,
-#             maximum=1000,
-#             step=1,
-#             value=50,
-#         ),
-#         gr.Slider(
-#             label="Repetition penalty",
-#             minimum=1.0,
-#             maximum=2.0,
-#             step=0.05,
-#             value=1.2,
-#         ),
-#     ],
-#     stop_btn=None,
-#     examples=[
-#         ["Ninawezaje kupata usingizi haraka?"],
-#         ["Bosi wangu anadhibiti sana, nifanye nini?"],
-#         ["Je, ni vipindi gani muhimu katika historia vya kujua kuvihusu?"],
-#         ["Ni kazi gani nzuri ikiwa ninataka kupata pesa lakini pia kufurahiya?"],
-#         ["Nivae nini kwenye harusi?"],
-#     ],
-# )
-
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Enter your message")
-    submit_btn = gr.Button("Submit")
-    clear = gr.Button("Clear")
+    msg = gr.Textbox(label="Ingiza ujumbe wako / Enter your message")
+    submit_btn = gr.Button("Wasilisha / Submit")
+    clear = gr.Button("Wazi / Clear")
 
     def user(user_message, history):
         return "", history + [[user_message, None]]
@@ -163,9 +112,9 @@ with gr.Blocks(css="style.css") as demo:
             history[-1][1] = bot_message
             yield history
 
-    gr.Examples(examples=examples, inputs=[msg], label="Examples")
+    gr.Examples(examples=examples, inputs=[msg], label="Mifano / Examples")
 
-    with gr.Accordion("Advanced Options", open=False):
+    with gr.Accordion("Chaguzi za Juu / Advanced Options", open=False):
         max_new_tokens = gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS)
         temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6)
         top_p = gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9)