Update app.py

app.py

@@ -11,7 +11,7 @@ class XylariaChat:
 
         # Initialize the inference client
         self.client = InferenceClient(
-            model="Qwen/Qwen-72B-Chat",  # Changed model name 
+            model="Qwen/Qwen-72B-Chat",  # Changed model name
             token=self.hf_token
         )
 
@@ -200,14 +200,14 @@ Capabilities:
                 clear = gr.Button("Clear Conversation")
                 clear_memory = gr.Button("Clear Memory")
 
-            # Load chat history from local storage on page load
+            # Use `gr.State` to manage initial chatbot value and `demo.load` for initialization
+            initial_chat_history = gr.State([])
+
             demo.load(
-                fn=None,
-                _js=load_from_local_storage_js  # Pass the JS function as a string
-            ).then(
-                fn=lambda x: x,
-                inputs=chatbot,
-                outputs=chatbot
+              fn=lambda: initial_chat_history.value,
+              inputs=None,
+              outputs=[chatbot],
+              _js=load_from_local_storage_js
             )
 
             # Submit functionality with local storage save