Update app.py

app.py

@@ -7,7 +7,7 @@ app = FastAPI(title="MedGemma ICD-10 Remote API")
 
 HF_TOKEN = os.getenv("HF_TOKEN", "YOUR_HF_API_TOKEN")
 MODEL_ID = "google/medgemma-4b-it"
-API_URL = f"https://router.huggingface.co/hf-inference/v1/models/{MODEL_ID}"
+API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
 
 class TextInput(BaseModel):
@@ -20,12 +20,14 @@ def root():
 @app.post("/predict")
 def predict_icd10(input: TextInput):
     try:
+        # dummy image
         dummy = Image.fromarray(np.zeros((224, 224, 3), dtype=np.uint8))
         buf = io.BytesIO()
         dummy.save(buf, format="PNG")
         image_b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
 
         prompt = f"<image>\nConvert this medical note into ICD-10 code only:\n{input.text}\nICD-10 code:"
+
         payload = {"inputs": {"image": image_b64, "prompt": prompt}}
 
         r = requests.post(API_URL, headers=HEADERS, json=payload, timeout=90)
@@ -35,6 +37,6 @@ def predict_icd10(input: TextInput):
                 return {"icd10_code": data[0]["generated_text"].strip()}
             return {"raw_response": data}
         else:
-            return {"error": f"Hugging Face Router error {r.status_code}", "details": r.text}
+            return {"error": f"Hugging Face API error {r.status_code}", "details": r.text}
     except Exception as e:
         return {"error": str(e)}