Update app.py

app.py

@@ -56,10 +56,13 @@ def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice,seed):
 
 def format_prompt(message, history,seed):
     prompt = "<s>"
+    t=False
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
         prompt += f" {bot_response}</s> "
-    if user_prompt == message:
+        if user_prompt == message:
+            t=True
+    if t:        
         prompt == "<s>"+f"[INST] {message} [/INST]"
         return prompt
     else: