Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,12 +12,14 @@ qa_generator = pipeline("text2text-generation", model="google/flan-t5-base", dev
|
|
| 12 |
def extract_text_from_pdf(pdf_file):
|
| 13 |
"""Extract text from uploaded PDF"""
|
| 14 |
try:
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
|
|
|
|
|
|
| 21 |
return text[:15000] # Limit tokens
|
| 22 |
except Exception as e:
|
| 23 |
return f"Error reading PDF: {str(e)}"
|
|
|
|
| 12 |
def extract_text_from_pdf(pdf_file):
|
| 13 |
"""Extract text from uploaded PDF"""
|
| 14 |
try:
|
| 15 |
+
# Open the file path directly
|
| 16 |
+
with open(pdf_file.name, 'rb') as f:
|
| 17 |
+
pdf_reader = PyPDF2.PdfReader(f)
|
| 18 |
+
text = ""
|
| 19 |
+
# Limit to first 10 pages for CPU performance
|
| 20 |
+
max_pages = min(10, len(pdf_reader.pages))
|
| 21 |
+
for page_num in range(max_pages):
|
| 22 |
+
text += pdf_reader.pages[page_num].extract_text()
|
| 23 |
return text[:15000] # Limit tokens
|
| 24 |
except Exception as e:
|
| 25 |
return f"Error reading PDF: {str(e)}"
|