Run olmOCR with Qwen2_5_VLForConditionalGeneration

app.py

@@ -50,7 +50,7 @@ except Exception as e:
 # Load olmOCR
 try:
     PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
-    MODELS["olmOCR"] = Qwen2VLForConditionalGeneration.from_pretrained(
+    MODELS["olmOCR"] = Qwen2_5_VLForConditionalGeneration.from_pretrained(
         "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
     )
     PIPELINES["olmOCR"] = pipeline(