Update app.py

app.py

@@ -12,13 +12,14 @@ from loguru import logger
 from datetime import datetime
 from pathlib import Path
 import torch
-from transformers import AutoProcessor, Qwen2VLForConditionalGeneration
+from transformers import AutoProcessor, Qwen2VLForConditionalGeneration, Qwen2_5_VLForConditionalGeneration
+from transformers.image_utils import load_image
+
 from pdf2image import convert_from_path
 import html2text
 import markdown
 import tempfile
 
-
 pdf_suffixes = [".pdf"]
 image_suffixes = [".png", ".jpeg", ".jpg"]
 
@@ -39,7 +40,7 @@ with open(header_path, 'r') as header_file:
 device = "cuda" if torch.cuda.is_available() else "cpu"
 MODEL_ID = "Logics-MLLM/Logics-Parsing"
 processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
-model = Qwen2VLForConditionalGeneration.from_pretrained(
+model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID,
     trust_remote_code=True,
     torch_dtype=torch.float16 if device == "cuda" else torch.float32