Update app.py

app.py

@@ -7,19 +7,19 @@ from huggingface_hub import login
 login(token = os.environ['HF_TOKEN'])
 
 repo = Repository(
-    local_dir="chatbot_function",
+    local_dir="agent_function",
     repo_type="dataset",
     clone_from=os.environ['DATASET'],
     token=True
 )
 repo.git_pull()
 
-from chatbot_function.agent import ask
+from agent_function.agent import Agent
 
 
 # Streamed response emulator
 def response_generator(query):
-    ans = ask(query)
+    ans = Agent().ask(query)
     
     for word in ans.split(' '):
             yield word + " "