xavierbarbier commited on
Commit
373805e
·
verified ·
1 Parent(s): d7905bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -64,6 +64,7 @@ doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/
64
 
65
  reader = PdfReader( doc_path )
66
 
 
67
  text = []
68
  for p in np.arange(0, len(reader.pages), 1):
69
  page = reader.pages[int(p)]
@@ -80,7 +81,7 @@ text_embeddings = np.array([get_text_embedding(chunk) for chunk in chunks])
80
  d = text_embeddings.shape[1]
81
  index = faiss.IndexFlatL2(d)
82
  index.add(text_embeddings)
83
-
84
 
85
  def extract_text(file):
86
 
@@ -120,7 +121,7 @@ def qa(question):
120
  return prompt
121
 
122
  def test_func(text):
123
- return len(text_embeddings)
124
 
125
  with gr.Blocks() as demo:
126
 
 
64
 
65
  reader = PdfReader( doc_path )
66
 
67
+ """
68
  text = []
69
  for p in np.arange(0, len(reader.pages), 1):
70
  page = reader.pages[int(p)]
 
81
  d = text_embeddings.shape[1]
82
  index = faiss.IndexFlatL2(d)
83
  index.add(text_embeddings)
84
+ """
85
 
86
  def extract_text(file):
87
 
 
121
  return prompt
122
 
123
  def test_func(text):
124
+ return doc_path , len(reader.pages)
125
 
126
  with gr.Blocks() as demo:
127