fix requirements and minor fixes

app.py

@@ -31,6 +31,8 @@ def process_pdfs(parent_dir: Union[str,list]):
     if type(parent_dir) == str :
         parent_dir = [parent_dir]
     for file_path in parent_dir:
+        if ".pdf" not in file_path : # skip non pdf files
+            continue
         # creating a pdf file object
         pdfFileObj = open(file_path, 'rb')
         
@@ -91,6 +93,7 @@ def predict(query,file_paths, k=3):
 
 with gr.Blocks() as demo : 
     with gr.Column(): 
+        gr.Markdown("## PDF Search Engine")
         files = gr.Files(label="Upload PDFs",type="filepath",file_count="multiple")
         query = gr.Text(label="query")
         with gr.Accordion():

requirements.txt

@@ -1,3 +1,4 @@
 datasets
 PyPDF2
-torch
+torch
+transformers