medgemma_api.py

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"]