Update app.py

app.py

@@ -1,11 +1,24 @@
-import spaces
+import subprocess
+import os
 import gradio as gr
 from pdf2image import convert_from_path
 from byaldi import RAGMultiModalModel
 from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 import torch
-import subprocess
+
+# Run the commands from setup.sh to install poppler-utils
+def install_poppler():
+    try:
+        subprocess.run(["pdfinfo"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    except FileNotFoundError:
+        print("Poppler not found. Installing...")
+        # Run the setup commands
+        subprocess.run("apt-get update", shell=True)
+        subprocess.run("apt-get install -y poppler-utils", shell=True)
+
+# Call the Poppler installation check
+install_poppler()
 
 # Install flash-attn if not already installed
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
@@ -16,7 +29,7 @@ model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instru
                                                         trust_remote_code=True, torch_dtype=torch.bfloat16).cuda().eval()
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True)
 
-@spaces.GPU()
+@gr.spaces.GPU()
 def process_pdf_and_query(pdf_file, user_query):
     # Convert the PDF to images
     images = convert_from_path(pdf_file.name)  # pdf_file.name gives the file path