Update app.py

app.py

@@ -58,8 +58,8 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
     stop_tokens = [tokenizer.eos_token_id]
     instruction = system_prompt + "\n\n"
     for user, assistant in history:
-        instruction += f"User: {user}\nAssistant: {assistant}\n"
-    instruction += f"User: {message}\nAssistant:"
+        instruction += f"role:user, content: {user}\nrole:assistant, content: {assistant}\n"
+    instruction += f"role:user, content: {message}\nassistant:"
     
     print(instruction)
     
@@ -123,5 +123,5 @@ gr.ChatInterface(
         gr.Slider(0, 2, 1.1, label="Repetition penalty"),
         gr.Slider(0, 1, 0.95, label="Top P sampling"),
     ],
-    theme=gr.themes.Soft(primary_hue=COLOR),
+    #theme=gr.themes.Soft(primary_hue=COLOR),
 ).queue().launch()