push

tplbot/initializer.py

@@ -20,13 +20,13 @@ generation_model3 = None
 def initialize():
     global hf_client, index, docs, intent_clf, intent_le, generation_model, generation_model2, generation_model3
 
-    hf_client = SentenceTransformer('all-mpnet-base-v2')
-    index = faiss.read_index('data/tpl_rag_index_h1.faiss')
-    with open('data/tpl_rag_docs_h1.pkl','rb') as f:
+    hf_client = SentenceTransformer('models/all-mpnet-base-v2/all_mpnet_base_v2')
+    index = faiss.read_index('tpl_rag_index_h1.faiss')
+    with open('tpl_rag_docs_h1.pkl','rb') as f:
         docs = pickle.load(f)
-    with open('data/intent_clf.pkl','rb') as f:
+    with open('intent_clf.pkl','rb') as f:
         intent_clf = pickle.load(f)
-    with open('data/intent_le.pkl','rb') as f:
+    with open('intent_le.pkl','rb') as f:
         intent_le = pickle.load(f)
 
     api_key = os.environ.get('GEMINI_API_KEY')