Spaces:

prithivMLmods
/

Multimodal-OCR

Running on Zero

prithivMLmods commited on Apr 22

Commit

29eb5ae

verified ·

1 Parent(s): 1372f45

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ model_a = AutoModelForImageTextToText.from_pretrained(
     torch_dtype=torch.float16
 ).to(device).eval()
-MODEL_ID_W = "prithivMLmods/Garnet-OCR-7B-0422"
 processor_w = AutoProcessor.from_pretrained(MODEL_ID_W, trust_remote_code=True)
 model_w = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_W,

     torch_dtype=torch.float16
 ).to(device).eval()
+MODEL_ID_W = "allenai/olmOCR-7B-0725"
 processor_w = AutoProcessor.from_pretrained(MODEL_ID_W, trust_remote_code=True)
 model_w = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_W,