Update models/llamaCustom.py

models/llamaCustom.py

@@ -54,7 +54,7 @@ model_pipeline = pipeline(
     do_sample=True,
     top_p=0.95,
     top_k=50,
-    temperature=0.7,
+    temperature=0.0,
 )
 
 
@@ -101,8 +101,8 @@ class LlamaCustom:
             #     index = pickle.loads(file.readlines())
             return index
         else:
-            # documents = prepare_data(r"./assets/regItems.json")
-            documents = SimpleDirectoryReader(input_dir="./assets/pdf").load_data()
+            documents = prepare_data(r"./assets/regItems.json")
+            #documents = SimpleDirectoryReader(input_dir="./assets/pdf").load_data()
 
             index = GPTVectorStoreIndex.from_documents(
                 documents, service_context=self.service_context