sergiopaniego HF Staff commited on
Commit
8507438
·
verified ·
1 Parent(s): 7fd0db3

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +1 -1
retriever.py CHANGED
@@ -30,7 +30,7 @@ class GuestInfoRetrieverTool(Tool):
30
 
31
  def load_guest_dataset():
32
  # Load the dataset
33
- guest_dataset = datasets.load_dataset("sergiopaniego/unit3-invitees", split="train")
34
 
35
  # Convert dataset entries into Document objects
36
  docs = [
 
30
 
31
  def load_guest_dataset():
32
  # Load the dataset
33
+ guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
34
 
35
  # Convert dataset entries into Document objects
36
  docs = [