Update app.py

app.py

@@ -33,22 +33,20 @@ def invoke(openai_api_key, prompt, rag_option):
         raise gr.Error("Retrieval-Augmented Generation is required.")
 
     with lock:
-        #os.environ["OPENAI_API_KEY"] = openai_api_key
-
-        ###
         prompt = """
         I want to stay in a place that's warm and friendly, 
         and not too far from resturants, can you recommend a place? 
         Include a reason as to why you've chosen your selection.
         """
+
         print("111")
         dataset = load_dataset("MongoDB/airbnb_embeddings", streaming=True, split="train")
-        dataset = dataset.take(100)
+        #dataset = dataset.take(100)
         print("222")
         # Convert the dataset to a pandas dataframe
         dataset_df = pd.DataFrame(dataset)
-        dataset_df.head(5)
-        print("Columns:", dataset_df.columns)
+        #dataset_df.head(5)
+        #print("Columns:", dataset_df.columns)
         
         listings = process_records(dataset_df)
         print("333")