Update app.py

app.py

@@ -1,5 +1,4 @@
 import spaces
-import subprocess
 import os
 import gradio as gr
 from pdf2image import convert_from_path
@@ -7,6 +6,8 @@ from byaldi import RAGMultiModalModel
 from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 import torch
+import torchvision
+import subprocess
 
 # Run the commands from setup.sh to install poppler-utils
 def install_poppler():
@@ -30,7 +31,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)
 
-@gr.spaces.GPU()
+@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