freQuensy23 commited on
Commit
72f7cb5
1 Parent(s): 22865c0
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -36,9 +36,7 @@ def handle_files_and_query(query, files, chunk_overlap=50, token_per_chunk=256,
36
  for file in files:
37
  documents.extend(
38
  PyMuPDFLoader(file).
39
- load_and_split(SentenceTransformersTokenTextSplitter(model_name=model,
40
- chunk_overlap=chunk_overlap,
41
- tokens_per_chunk=token_per_chunk)))
42
  retriever = BM25Retriever.from_documents(documents, k=bm_25_answers)
43
  results += "Index created successfully!\n"
44
  print("Index created successfully!")
@@ -71,7 +69,7 @@ interface = gr.Interface(
71
  fn=handle_files_and_query,
72
  inputs=[
73
  gr.Textbox(lines=1, label="Enter your search query here..."),
74
- gr.File(file_count="multiple", type="filepath", file_types=[".pdf"], label="Upload a file here."),
75
  gr.Slider(minimum=1, maximum=100, value=50, label="Chunk Overlap"),
76
  gr.Slider(minimum=64, maximum=512, value=256, label="Tokens Per Chunk (чем больше - тем бОльшие куски книги "
77
  "сможем находить)"),
 
36
  for file in files:
37
  documents.extend(
38
  PyMuPDFLoader(file).
39
+ load_and_split(SentenceTransformersTokenTextSplitter()))
 
 
40
  retriever = BM25Retriever.from_documents(documents, k=bm_25_answers)
41
  results += "Index created successfully!\n"
42
  print("Index created successfully!")
 
69
  fn=handle_files_and_query,
70
  inputs=[
71
  gr.Textbox(lines=1, label="Enter your search query here..."),
72
+ gr.File(file_count="multiple", type="file", file_types=[".pdf"], label="Upload a file here."),
73
  gr.Slider(minimum=1, maximum=100, value=50, label="Chunk Overlap"),
74
  gr.Slider(minimum=64, maximum=512, value=256, label="Tokens Per Chunk (чем больше - тем бОльшие куски книги "
75
  "сможем находить)"),