Tuana commited on
Commit
d786d2c
·
1 Parent(s): 3f73a9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -18,10 +18,7 @@ preprocessor = PreProcessor(
18
  def pdf_to_document_store(pdf_files):
19
  document_store.delete_documents()
20
  converter = PDFToTextConverter(remove_numeric_tables=True, valid_languages=["en"])
21
- documents = []
22
- for pdf in pdf_files:
23
- logging.info(pdf.name)
24
- documents.append(converter.convert(file_path=pdf.name, meta=None))
25
  preprocessed_docs = preprocessor.process(documents)
26
  document_store.write_documents(preprocessed_docs)
27
  return None
@@ -34,7 +31,7 @@ def summarize(files):
34
 
35
  title = "Summarize one or more PDFs with a Haystack Summariser pipeline"
36
  iface = gr.Interface(fn=summarize,
37
- inputs="files",
38
  outputs="text",
39
  title=title,
40
  theme="default")
 
18
  def pdf_to_document_store(pdf_files):
19
  document_store.delete_documents()
20
  converter = PDFToTextConverter(remove_numeric_tables=True, valid_languages=["en"])
21
+ documents = [converter.convert(file_path=pdf.name, meta=None)]
 
 
 
22
  preprocessed_docs = preprocessor.process(documents)
23
  document_store.write_documents(preprocessed_docs)
24
  return None
 
31
 
32
  title = "Summarize one or more PDFs with a Haystack Summariser pipeline"
33
  iface = gr.Interface(fn=summarize,
34
+ inputs=[gr.inputs.File(file_count="single",type="file", label="Upload a pdf")],
35
  outputs="text",
36
  title=title,
37
  theme="default")