Update models/llamaCustom.py

models/llamaCustom.py

@@ -87,7 +87,7 @@ class LlamaCustom:
     def __init__(self, name: str) -> None:
         self.vector_index = self.initialize_index(index_name=name)
 
-    def prepare_data(file_path: str):
+    def prepare_data(self, file_path: str):
         df = pd.read_json(file_path)
         df = df.replace(to_replace="", value=np.nan).dropna(axis=0)  # remove null values
     
@@ -122,7 +122,7 @@ class LlamaCustom:
             #     index = pickle.loads(file.readlines())
             return index
         else:
-            documents = prepare_data(r"./assets/regItems.json")
+            documents = self.prepare_data(r"./assets/regItems.json")
             #documents = SimpleDirectoryReader(input_dir="./assets/pdf").load_data()
 
             index = GPTVectorStoreIndex.from_documents(