Update app.py

app.py

@@ -64,7 +64,7 @@ doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/
 
 reader = PdfReader( doc_path )
 
-"""
+
 text = []
 for p in np.arange(0, len(reader.pages), 1):
     page = reader.pages[int(p)]
@@ -81,7 +81,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)
-"""
+
 
 def extract_text(file):
 
@@ -121,7 +121,7 @@ def qa(question):
     return prompt
 
 def test_func(text):
-    return reader.pages[0].extract_text()
+    return len(text_embeddings)
 
 with gr.Blocks() as demo: