Update app.py

app.py

@@ -4,6 +4,20 @@ from transformers import AutoModel, AutoTokenizer
 from pdf2image import convert_from_path
 import torch
 import os
+import os
+import subprocess
+import streamlit as st
+import google.generativeai as genai
+from transformers import AutoModel, AutoTokenizer
+from pdf2image import convert_from_path
+import torch
+
+# Ensure Poppler is installed
+def install_poppler():
+    if not os.path.exists("/usr/bin/pdftoppm"):  # Check if Poppler is installed
+        st.warning("Installing Poppler for PDF processing...")
+        subprocess.run(["apt-get", "update"])
+        subprocess.run(["apt-get", "install", "-y", "poppler-utils"])
 
 # Load the OCR model
 tokenizer = AutoTokenizer.from_pretrained("ucaslcl/GOT-OCR2_0", trust_remote_code=True)