ocr-time-machine

Running on Zero

storytracer commited on Aug 29

Commit

15f06e4

unverified ·

1 Parent(s): 64747fe

Updated olmOCR to latest version

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ except Exception as e:
 try:
     PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
     MODELS["olmOCR"] = Qwen2VLForConditionalGeneration.from_pretrained(
-        "allenai/olmOCR-7B-0225-preview", torch_dtype=torch.bfloat16, device_map="auto"
     )
     PIPELINES["olmOCR"] = pipeline(
         "image-text-to-text", model=MODELS["olmOCR"], processor=PROCESSORS["olmOCR"]

 try:
     PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
     MODELS["olmOCR"] = Qwen2VLForConditionalGeneration.from_pretrained(
+        "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
     )
     PIPELINES["olmOCR"] = pipeline(
         "image-text-to-text", model=MODELS["olmOCR"], processor=PROCESSORS["olmOCR"]