cowcow02 commited on
Commit
e9e8863
1 Parent(s): 5543281

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +6 -3
train.py CHANGED
@@ -11,10 +11,13 @@ openai.api_key = OPENAI_API_KEY
11
 
12
  print('Start Loading Data ...')
13
 
14
- PagedCSVReader = download_loader("PagedCSVReader")
 
 
15
 
16
- loader = PagedCSVReader(encoding="utf-8")
17
- documents = loader.load_data(file=Path('train-assets/training-target-simple.csv'))
 
18
 
19
  pinecone.init(
20
  api_key=PINECONE_API_KEY,
 
11
 
12
  print('Start Loading Data ...')
13
 
14
+ # PagedCSVReader = download_loader("PagedCSVReader")
15
+ # loader = PagedCSVReader(encoding="utf-8")
16
+ # documents = loader.load_data(file=Path('train-assets/training-target-simple.csv'))
17
 
18
+ PDFReader = download_loader("PDFReader")
19
+ loader = PDFReader()
20
+ documents = loader.load_data(file=Path('./train-assets/faq.pdf'))
21
 
22
  pinecone.init(
23
  api_key=PINECONE_API_KEY,