Spaces:

prithivMLmods
/

Multimodal-OCR3

Running on Zero

prithivMLmods commited on Oct 17

Commit

0571fd6

verified ·

1 Parent(s): 8298b76

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,8 +133,8 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     #_attn_implementation="flash_attention_2"
 ).to(device).eval()
-# Load Dots.OCR
-MODEL_ID_D = "rednote-hilab/dots.ocr"
 processor_d = AutoProcessor.from_pretrained(MODEL_ID_D, trust_remote_code=True)
 model_d = AutoModelForCausalLM.from_pretrained(
     MODEL_ID_D,

     #_attn_implementation="flash_attention_2"
 ).to(device).eval()
+# Load Dots.OCR (rednote-hilab/dots.ocr)
+MODEL_ID_D = "strangervisionhf/dot.fix"
 processor_d = AutoProcessor.from_pretrained(MODEL_ID_D, trust_remote_code=True)
 model_d = AutoModelForCausalLM.from_pretrained(
     MODEL_ID_D,