zhtet commited on
Commit
6db6d08
1 Parent(s): 459520e

Update models/llamaCustom.py

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +1 -2
models/llamaCustom.py CHANGED
@@ -87,8 +87,7 @@ class LlamaCustom:
87
  def __init__(self, model_name: str) -> None:
88
  self.vector_index = self.initialize_index(model_name=model_name)
89
 
90
- @st.cache_resource
91
- def initialize_index(_self, model_name: str):
92
  index_name = model_name.split("/")[-1]
93
 
94
  file_path = f"./vectorStores/{index_name}"
 
87
  def __init__(self, model_name: str) -> None:
88
  self.vector_index = self.initialize_index(model_name=model_name)
89
 
90
+ def initialize_index(self, model_name: str):
 
91
  index_name = model_name.split("/")[-1]
92
 
93
  file_path = f"./vectorStores/{index_name}"