Spaces:

prithivMLmods
/

POINTS-Reader-OCR

Running on Zero

prithivMLmods commited on Nov 1

Commit

e52ae93

verified ·

1 Parent(s): ce5efe6

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -117,12 +117,11 @@ print("Using device:", device)
 # --- Model Loading: tencent/POINTS-Reader ---
-MODEL_PATH = 'strangervisionhf/points-reader-sdpa'
 print(f"Loading model: {MODEL_PATH}")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_PATH,
-    #attn_implementation="eager",
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
     device_map='auto'

 # --- Model Loading: tencent/POINTS-Reader ---
+MODEL_PATH = 'tencent/POINTS-Reader'
 print(f"Loading model: {MODEL_PATH}")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_PATH,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
     device_map='auto'