File size: 460 Bytes
bd1f144 9ed9659 d058706 9ed9659 d058706 9ed9659 d058706 9ed9659 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
import requests
import base64
from transformers import pipeline
pipe = pipeline("image-text-to-text", model="google/medgemma-4b-it")
def query_medgemma_local(image_path, question):
messages = [
{
"role": "user",
"content": [
{"type": "image", "source": image_path},
{"type": "text", "text": question}
]
}
]
return pipe(messages)[0]["generated_text"]
|