Update app.py

app.py

@@ -32,13 +32,12 @@ def format_prompt_default(message, history,cust_p):
 def format_prompt_gemma(message, history,cust_p):
     prompt = ""
     if history:
-        #<start_of_turn>userHow does the brain work?<end_of_turn><start_of_turn>model
         for user_prompt, bot_response in history:
-            prompt += f"{user_prompt}\n"
-            print(prompt)
-            prompt += f"{bot_response}\n"
-            print(prompt)
-    #prompt += f"<start_of_turn>user{message}<end_of_turn><start_of_turn>model"
+            prompt += f"<start_of_turn>user{user_prompt}<end_of_turn>"
+            prompt += f"<start_of_turn>model{bot_response}<end_of_turn>"
+            if VERBOSE==True:
+                print(prompt)
+    #prompt += f"<start_of_turn>user\n{message}<end_of_turn>\n<start_of_turn>model\n"
     prompt+=cust_p.replace("USER_INPUT",message)
     return prompt
 def format_prompt_openc(message, history,cust_p):
@@ -68,7 +67,7 @@ def format_prompt_mixtral(message, history,cust_p):
     return prompt
 
 def format_prompt_choose(message, history, cust_p, model_name):
-    if "gemma" in models[model_name].lower() and "it" in models[model_name].lower():
+    if "gemma" in models[model_name].lower():
         return format_prompt_gemma(message,history,cust_p)
     if "mixtral" in models[model_name].lower():
         return format_prompt_mixtral(message,history,cust_p)