Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 # Function to extract text from PDF
 def extract_text_from_pdf(pdf_path):
     text = ""
-    document = fitz.open(pdf_path)
+    document = fitz.open(pdf_path.name)
     for page_num in range(len(document)):
         page = document.load_page(page_num)
         text += page.get_text()
@@ -22,7 +22,7 @@ qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
 # Function to answer questions based on the PDF content
 def answer_question(pdf_file, question):
     # Extract text from the uploaded PDF file
-    content = extract_text_from_pdf(pdf_file.name)
+    content = extract_text_from_pdf(pdf_file)
     # Get the answer using the question-answering pipeline
     result = qa_pipeline(question=question, context=content)
     return result['answer']
@@ -30,7 +30,7 @@ def answer_question(pdf_file, question):
 # Define the Gradio interface
 iface = gr.Interface(
     fn=answer_question,
-    inputs=[gr.inputs.File(file_count="single", type="file"), gr.inputs.Textbox(lines=2, placeholder="Ask a question...")],
+    inputs=[gr.File(label="PDF File", file_types=[".pdf"]), gr.Textbox(lines=2, placeholder="Ask a question...")],
     outputs="text",
     title="DistilBERT Question Answering",
     description="Upload a PDF and ask questions based on the content of the PDF."