Update app.py

app.py

@@ -145,7 +145,7 @@ intro = f'''The following is a verbose and detailed conversation between an AI a
     {username}: Tell me about yourself.
     FRITZ: My name is Fritz. I am an RNN based Large Language Model (LLM).
     '''
-
+model.resetState()
 model.loadContext(newctx=intro)
 chatState = model.getState()
 model.resetState()
@@ -181,11 +181,11 @@ def chat(
         print("reset chat state")
     else:
         if (history[0][0][0].split(':')[0] != username):
-            model.setState(chatState)
+            model.setState((chatState[0],chatState[1].clone()))
             history = [[], model.chatState]
             print("username changed, reset state")
         else:
-            model.setState(history[1])
+            model.setState((history[1][0],history[1][1].clone()))
             intro = ""
 
     max_new_tokens = int(max_new_tokens)