Chris4K commited on
Commit
f8e6299
·
verified ·
1 Parent(s): 3e98f5c

Update controller.py

Browse files
Files changed (1) hide show
  1. controller.py +1 -0
controller.py CHANGED
@@ -33,6 +33,7 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
33
  return response
34
 
35
  def handle_submission_chat(user_message, response):
 
36
  agent_chat_bot =get_conversation_chain()
37
  text = agent_chat_bot.predict(input=user_message, context=response)
38
  print(text)
 
33
  return response
34
 
35
  def handle_submission_chat(user_message, response):
36
+ HUGGINGFACEHUB_API_TOKEN = os.environ['HF_token']
37
  agent_chat_bot =get_conversation_chain()
38
  text = agent_chat_bot.predict(input=user_message, context=response)
39
  print(text)