Spaces:

aarishshahmohsin
/

ocr_reader

Running

App Files Files Community

ocr_reader / app copy.py

aarishshahmohsin's picture

aarishshahmohsin

added aarish model

aafffbc 10 days ago

history blame contribute delete

No virus

760 Bytes

	from transformers import AutoTokenizer, AutoModel
	# tokenizer = AutoTokenizer.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, device_map='cpu')
	# model = AutoModel.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
	tokenizer = AutoTokenizer.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, device_map='cpu')
	model = AutoModel.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)

	model = model.eval().cpu()

	image_path = './image.png'
	english_extraction = model.chat(tokenizer, image_path, ocr_type='ocr')
	print(english_extraction)