Create medgemma_api.py

medgemma_api.py

@@ -0,0 +1,27 @@
+import requests
+import base64
+
+# 你需要在 Hugging Face Space 的 Secrets 里配置一个名为 HUGGINGFACE_TOKEN 的值
+HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/google/medgemma-4b-it"
+HEADERS = {"Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}"}
+
+def query_medgemma(image_path, question):
+    with open(image_path, "rb") as f:
+        image_bytes = f.read()
+
+    encoded_image = base64.b64encode(image_bytes).decode("utf-8")
+    payload = {
+        "inputs": [
+            {
+                "role": "user",
+                "content": [
+                    {"type": "image", "image": encoded_image},
+                    {"type": "text", "text": question}
+                ]
+            }
+        ]
+    }
+
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    return response.json()[0]["generated_text"] if response.ok else "Error: Unable to get response."