wt002 commited on
Commit
ac77e39
·
verified ·
1 Parent(s): 65fbb17

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +6 -2
agent.py CHANGED
@@ -136,8 +136,12 @@ sys_msg = SystemMessage(content=system_prompt)
136
  # -------------------------------
137
  # Step 1: Load documents from CSV file (max 165 rows)
138
  # -------------------------------
139
- csv_file_path = "/home/wendy/Downloads/documents.csv" # Replace with your actual file path
140
- df = pd.read_csv(csv_file_path).head(165)
 
 
 
 
141
 
142
  # Check if 'content' column exists
143
  assert 'content' in df.columns, "'content' column is required in the CSV file."
 
136
  # -------------------------------
137
  # Step 1: Load documents from CSV file (max 165 rows)
138
  # -------------------------------
139
+
140
+ # -------------------------------
141
+ # Step 1: Load documents from CSV URL (max 165 rows)
142
+ # -------------------------------
143
+ csv_url = "https://huggingface.co/spaces/wt002/Final_Assignment_Project/blob/main/documents.csv" # Replace with your actual URL
144
+ df = pd.read_csv(csv_url).head(165)
145
 
146
  # Check if 'content' column exists
147
  assert 'content' in df.columns, "'content' column is required in the CSV file."