pulseDemo / create_index.py
svummidi's picture
Update create_index.py
1c65feb
from llama_index import GPTSimpleVectorIndex, SimpleDirectoryReader
from pathlib import Path
from llama_index import download_loader
import os
os.environ["OPENAI_API_KEY"] = "REPLACE"
PandasCSVReader = download_loader("PandasCSVReader")
loader = PandasCSVReader()
documents = loader.load_data(file=Path('./input.csv'))
index = GPTSimpleVectorIndex(documents)
index.save_to_disk('./output.json')