Update app.py

app.py

@@ -26,7 +26,11 @@ def generate(prompt,history,role="ASSISTANT"):
         history=[]
     seed = random.randint(1,9999999999999)
     print(seed)
-    system_prompt = prompts.role
+    if role = "ASSISTANT"
+        system_prompt = prompts.ASSISTANT
+    elif role = "RESEARCHER"
+        system_prompt = prompts.RESEARCHER
+    else: system_prompt = ""
     generate_kwargs = dict(
         temperature=0.9,
         max_new_tokens=512,