Update medgemma_api.py

medgemma_api.py

@@ -2,27 +2,19 @@ import os
 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}"}
+from transformers import pipeline
 
-def query_medgemma(image_path, question):
-    with open(image_path, "rb") as f:
-        image_bytes = f.read()
+pipe = pipeline("image-text-to-text", model="google/medgemma-4b-it")
 
-    encoded_image = base64.b64encode(image_bytes).decode("utf-8")
-    payload = {
-        "inputs": [
-            {
-                "role": "user",
-                "content": [
-                    {"type": "image", "image": encoded_image},
-                    {"type": "text", "text": question}
-                ]
-            }
-        ]
-    }
+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"]
 
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
-    return response.json()[0]["generated_text"] if response.ok else "Error: Unable to get response."