Update app.py

app.py

@@ -21,12 +21,12 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def generate(prompt,history):
+def generate(prompt,history,role="ASSISTANT"):
     if not history:
         history=[]
     seed = random.randint(1,9999999999999)
     print(seed)
-    system_prompt = prompts.ASSISTANT
+    system_prompt = prompts.role
     generate_kwargs = dict(
         temperature=0.9,
         max_new_tokens=512,