Update app.py

app.py

@@ -60,8 +60,10 @@ def get_text_embedding(text):
 
     return embeddings.embed_query(text)
 
-doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/NGAP 01042024.pdf", repo_type="space")
+doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/embeddings_ngap.faiss", repo_type="space")
 
+index = faiss.read_index(doc_path)
+"""
 reader = PdfReader( doc_path )
 
 
@@ -80,7 +82,7 @@ text_embeddings = np.array([get_text_embedding(chunk) for chunk in chunks])
 
 d = text_embeddings.shape[1]
 index = faiss.IndexFlatL2(d)
-index.add(text_embeddings)
+index.add(text_embeddings)"""
 
 
 def extract_text(file):
@@ -121,7 +123,7 @@ def qa(question):
     return prompt
 
 def test_func(text):
-    return len(text_embeddings)
+    return len(index)
 
 with gr.Blocks() as demo: