Docfile commited on
Commit
1685442
·
verified ·
1 Parent(s): c4d1308

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,10 +28,10 @@ template = """Utillisez les éléments de contexte suivants pour répondre à la
28
 
29
  User: {user_message}
30
 
31
- Chatbot:"""
32
 
33
  prompt = PromptTemplate(
34
- input_variables=["chat_history", "user_message", "dikota_content"], template=template
35
  )
36
 
37
  memory = ConversationBufferMemory(memory_key="chat_history")
@@ -43,7 +43,7 @@ llm_chain = LLMChain(
43
  memory=memory,
44
  )
45
 
46
- def get_text_response(user_message, history, dikota_content): # dikota_content added
47
  response = llm_chain.predict(user_message=user_message)
48
  print(user_message)
49
  return response
 
28
 
29
  User: {user_message}
30
 
31
+ Chatbot:""".format(dikota_content)
32
 
33
  prompt = PromptTemplate(
34
+ input_variables=["chat_history", "user_message"], template=template
35
  )
36
 
37
  memory = ConversationBufferMemory(memory_key="chat_history")
 
43
  memory=memory,
44
  )
45
 
46
+ def get_text_response(user_message, history): # dikota_content added
47
  response = llm_chain.predict(user_message=user_message)
48
  print(user_message)
49
  return response