Update app.py

app.py

@@ -1,4 +1,5 @@
-import fitz
+
+import pdfplumber
 import gradio as gr
 import torch
 
@@ -18,18 +19,17 @@ pipe = pipeline(
 print("Model loaded.")
 
 
-def extract_text(pdf_file):
 
+def extract_text(pdf_file):
     text = ""
 
-    doc = fitz.open(pdf_file.name)
-
-    for page in doc:
-        text += page.get_text()
-
-    doc.close()
+    with pdfplumber.open(pdf_file.name) as pdf:
+        for page in pdf.pages:
+            page_text = page.extract_text()
+            if page_text:
+                text += page_text + "\n"
 
-    return text[:20000]
+    return text[:12000]
 
 
 def build_prompt(policy_text):