Update app.py

app.py

@@ -192,7 +192,7 @@ def agent(prompt_in,history,mod=2):
         yield [{'role':'assistant','content':str(outph)}]
         role="MANAGER"
         outp=generate(prompt,history,mod,128,seed,role,in_data)
-        outp0=list(outp)[0].split('<|im_end}>')[0]
+        outp0=list(outp)[0].split('<|im_end|>')[0]
         #outp0 = re.sub('[^a-zA-Z0-9\s.,?!%()]', '', outpp)
         history=history+[{'role':'assistant','content':str(outp0)}]
         yield [{'role':'assistant','content':str(outp0)}]
@@ -201,7 +201,7 @@ def agent(prompt_in,history,mod=2):
                 try:
                     com_line = line.split('action:')[1]
                     fn = com_line.split('action_input=')[0]
-                    com = com_line.split('action_input=')[1]
+                    com = com_line.split('action_input=')[1].split('<|im_end|>')[0]
                     #com = com_line.split('action_input=')[1].replace('<|im_end|>','').replace("}","").replace("]","").replace("'","")
                     print(com)
                 except Exception as e: