Update app.py

app.py

@@ -95,7 +95,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
         raise ValueError("Chat mode and FIM are mutually exclusive. Choose one or the other.")
 
     if chat_mode:
-        chat_prompt = "Human: " + prompt + "\n\nAssistant:"
+        chat_prompt = "Human: " + prompt + "\n\nAssistant: "
         prompt = HHH_PROMPT + chat_prompt
     
     if FIM_INDICATOR in prompt:
@@ -111,7 +111,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
     if fim_mode:
         output = prefix
     elif chat_mode:
-        output = chat_prompt
+        output = prompt #chat_prompt
     else:
         output = prompt