Views
No views yet
1
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from PIL import Image
4
5model_id = "ayoubkirouane/moondream2-image-captcha"
6model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
7tokenizer = AutoTokenizer.from_pretrained(model_id , trust_remote_code=True)
8
9image = Image.open('/content/test.png')
10enc_image = model.encode_image(image)
11print(model.answer_question(enc_image, "What does the text say?", tokenizer))
12
13