ghengx commited on
Commit
457890e
·
1 Parent(s): b5cccbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,19 +7,19 @@ from huggingface_hub import login
7
  login(token = os.environ['HF_TOKEN'])
8
 
9
  repo = Repository(
10
- local_dir="chatbot_function",
11
  repo_type="dataset",
12
  clone_from=os.environ['DATASET'],
13
  token=True
14
  )
15
  repo.git_pull()
16
 
17
- from chatbot_function.agent import ask
18
 
19
 
20
  # Streamed response emulator
21
  def response_generator(query):
22
- ans = ask(query)
23
 
24
  for word in ans.split(' '):
25
  yield word + " "
 
7
  login(token = os.environ['HF_TOKEN'])
8
 
9
  repo = Repository(
10
+ local_dir="agent_function",
11
  repo_type="dataset",
12
  clone_from=os.environ['DATASET'],
13
  token=True
14
  )
15
  repo.git_pull()
16
 
17
+ from agent_function.agent import Agent
18
 
19
 
20
  # Streamed response emulator
21
  def response_generator(query):
22
+ ans = Agent().ask(query)
23
 
24
  for word in ans.split(' '):
25
  yield word + " "