Update app.py

app.py

@@ -11,18 +11,20 @@ from langchain_core.messages import SystemMessage
 from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain_groq import ChatGroq
 
-def generate_response(user_input, history, model, temperature, max_tokens, top_p, seed):
-    print( "Model =", model)
+# Initialize Groq Langchain chat object and conversation
+groq_chat = ChatGroq(
+        groq_api_key=os.environ.get("GROQ_API_KEY"),
+)
 
-    # Initialize Groq Langchain chat object and conversation
-    groq_chat = ChatGroq(
-            groq_api_key=os.environ.get("GROQ_API_KEY"),
-            model_name=model
-    )
+# Initialize memory to manages the chat history, 
+# ensuring the AI remembers the specified number of history messages, in this case 5.
+memory = ConversationBufferWindowMemory(k=5, memory_key="chat_history", return_messages=True)
 
-    # Manages the chat history, ensuring the AI remembers the specified number of history messages, in this case 5.
-    memory = ConversationBufferWindowMemory(k=5, memory_key="chat_history", return_messages=True)
 
+def generate_response(user_input, history, model, temperature, max_tokens, top_p, seed):
+    print( "Model =", model)
+    
+    groq_chat.model_name = model_name
     prompt = ChatPromptTemplate.from_messages(
         [
             # This is the persistent system prompt, sets the initial context for the AI.
@@ -63,7 +65,6 @@ example1 = [
             ["What to San Francisco?"],        
             ["Then what to Beijing?"],
             ["And what to Kyoto?"],
-            ["Who are you?"],    
         ]
 
 # Create the Gradio interface