Update app.py

app.py

@@ -96,9 +96,9 @@ api_response_prompt = PromptTemplate(
 def setup_multiple_chains():
     
     llm = HuggingFaceEndpoint(
-    repo_id="google/gemma-2-2b-it", #"norallm/normistral-7b-warm-instruct", 
+        #repo_id="google/gemma-2-2b-it", #"norallm/normistral-7b-warm-instruct", 
         #endpoint_url="http://localhost:8010/",
-        #model="gemma-2-2b-it",
+        model="normistral-7b-warm-instruct",
         max_new_tokens=512,
         top_k=10,
         top_p=0.95,
@@ -156,11 +156,11 @@ async def handle_message(message: cl.Message):
 
     if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):   # ex. "EQJLCQ362149"
         
-        response = await api_chain.acall(user_message, 
+        response = await api_chain.ainvoke(user_message, 
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
 
     else: 
-        response = await llm_chain.acall(user_message,
+        response = await llm_chain.ainvoke(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])