Update app.py

app.py

@@ -26,9 +26,9 @@ def generate(prompt,history,role="ASSISTANT"):
         history=[]
     seed = random.randint(1,9999999999999)
     print(seed)
-    if role = "ASSISTANT"
+    if role == "ASSISTANT"
         system_prompt = prompts.ASSISTANT
-    elif role = "RESEARCHER"
+    elif role == "RESEARCHER"
         system_prompt = prompts.RESEARCHER
     else: system_prompt = ""
     generate_kwargs = dict(
@@ -45,6 +45,9 @@ def generate(prompt,history,role="ASSISTANT"):
     buf = ""
     for response in stream:
         output += response.token.text
+        '''buf += response.token.text
+        if "\n" in buf or ". " in buf or "</s>" in buf:
+            yield [(prompt,output)]'''
         yield [(prompt,output)]
 
 def load_mod(model):