Update custom_utils.py

custom_utils.py

@@ -21,8 +21,8 @@ def connect_to_database():
     return db, collection
 
 def rag_ingestion(collection):
-    #dataset = load_dataset("bstraehle/airbnb-san-francisco-202403-embed", streaming=True, split="train")
-    dataset = load_dataset("bstraehle/test", streaming=True, split="train")
+    dataset = load_dataset("bstraehle/airbnb-san-francisco-202403-embed", streaming=True, split="train")
+    #dataset = load_dataset("bstraehle/test", streaming=True, split="train")
     #dataset_df = pd.DataFrame(dataset)
     #listings = process_records(dataset_df)
     collection.delete_many({})