Update app.py

app.py

@@ -31,13 +31,17 @@ def user(message, history):
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     history = history or []
     
-    # Criar o chat template
-    prompt_template=f'''<|im_start|>system
-    {system_message}<|im_end|>
-    <|im_start|>user
-    {prompt}<|im_end|>
-    <|im_start|>assistant
+    # A última mensagem do usuário
+    user_prompt = history[-1][0]
+    
+    # Definindo o template e o prompt
+    prompt_template = f'''system
+    {system_message.strip()}
+    user
+    {user_prompt}
+    assistant
     '''
+    
     # Preparando o input
     input_ids = tokenizer(prompt_template, return_tensors='pt').input_ids  # .cuda() se você estiver usando GPU