Update app.py

app.py

@@ -60,12 +60,11 @@ def format_prompt(message, history,seed):
         prompt += f"[INST] {user_prompt} [/INST]"
         prompt += f" {bot_response}</s> "
     if user_prompt == message:
-        print (user_prompt)
-        if seed != 0:
-            prompt += f"[INST] {message} [/INST]"
-            return prompt
-        else:
-            return prompt
+        prompt == "<s>"+f"[INST] {message} [/INST]"
+        return prompt
+    else:
+        prompt += f"[INST] {message} [/INST]"
+        return prompt
 
 def run_gpt(in_prompt,history,model_drop,seed):
     client = InferenceClient(c_models[int(model_drop)])