Update image_pipeline.py

image_pipeline.py

@@ -1,17 +1,17 @@
-from huggingface_hub import InferenceApi
+from transformers import AutoProcessor, AutoModel
 from PIL import Image
-import base64
+
 from config import HF_IMAGE_MODEL, HF_TOKEN
 
-# Initialize Hugging Face Inference API
-inference = InferenceApi(repo_id=HF_IMAGE_MODEL, token=HF_TOKEN)
+# Load the Hugging Face model for medical image analysis
+model = AutoModel.from_pretrained(HF_IMAGE_MODEL, trust_remote_code=True)
+processor = AutoProcessor.from_pretrained(HF_IMAGE_MODEL)
 
 def analyze_medical_image(image_file):
     """
-    Analyze a medical image using the Hugging Face Inference API.
+    Process and analyze a medical image to generate diagnostic insights.
     """
-    with open(image_file, "rb") as img:
-        base64_image = base64.b64encode(img.read()).decode("utf-8")
-
-    response = inference(inputs={"image": base64_image})
-    return response.get("generated_text", "No insights generated.")
+    image = Image.open(image_file).convert("RGB")
+    inputs = processor(images=image, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=256)
+    return processor.decode(outputs[0], skip_special_tokens=True)