Hugging Face's logo

Spaces:
prithivMLmods
/
Doc-VLMs-OCR
Running on Zero

App Files Community
1
prithivMLmods commited on
Commit
1e727a6
·
verified ·
1 Parent(s): 8ad61e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ model_t = Qwen2_5_VLForConditionalGeneration.from_pretrained(
60
  # Load olmOCR-7B-0225-preview
61
  MODEL_ID_O = "google/gemma-3n-E4B-it"
62
  processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
63
- model_o = Gemma3nForConditionalGeneration.from_pretrained(
64
  MODEL_ID_O,
65
  trust_remote_code=True,
66
  torch_dtype=torch.float16
 
60
  # Load olmOCR-7B-0225-preview
61
  MODEL_ID_O = "google/gemma-3n-E4B-it"
62
  processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
63
+ model_o = AutoModelForImageTextToText.from_pretrained(
64
  MODEL_ID_O,
65
  trust_remote_code=True,
66
  torch_dtype=torch.float16