bin20 commited on
Commit
6387c71
·
1 Parent(s): b96f319

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def get_csv_file(csv_file):
42
  temp_filepath = os.path.join(temp_dir.name, csv_file.name)
43
  with open(temp_filepath, "w", encoding="utf-8") as csvfile:
44
  csvfile.write(csv_file.getvalue().decode('utf-8'))
45
- csv_loader = CSVLoader(temp_filepath, text_column='text_column_name')
46
  csv_data = csv_loader.load()
47
  text_from_csv = '\n'.join(csv_data['text_column_name']) if 'text_column_name' in csv_data else ''
48
  return text_from_csv
 
42
  temp_filepath = os.path.join(temp_dir.name, csv_file.name)
43
  with open(temp_filepath, "w", encoding="utf-8") as csvfile:
44
  csvfile.write(csv_file.getvalue().decode('utf-8'))
45
+ csv_loader = CSVLoader(temp_filepath, text_column.name)
46
  csv_data = csv_loader.load()
47
  text_from_csv = '\n'.join(csv_data['text_column_name']) if 'text_column_name' in csv_data else ''
48
  return text_from_csv