MohamedTry/ICD

1from fastapi import FastAPI2from pydantic import BaseModel3import requests4import os5 6app = FastAPI(title="ICD-10 Remote Inference")7 8# 🔑 ضع توكنك من إعدادات Hugging Face (Read access)9HF_TOKEN = os.getenv("HF_TOKEN", "YOUR_HF_API_TOKEN")10 11# قائمة الموديلات المراد تجربتها12MODELS = [13    "AkshatSurolia/ICD-10-Code-Prediction",14    "rjac/biobert-ICD10-L3-mimic",15    "dataphysician/ModernBERT-icd10-classifier"16]17 18HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}19 20class TextInput(BaseModel):21    text: str22 23@app.get("/")24def home():25    return {"status": "running", "models_to_try": MODELS}26 27def query_model(model_name: str, text: str):28    url = f"https://api-inference.huggingface.co/models/{model_name}"29    payload = {"inputs": text}30    response = requests.post(url, headers=HEADERS, json=payload, timeout=30)31    if response.status_code == 200:32        return response.json()33    else:34        raise RuntimeError(f"{model_name} failed: {response.status_code}")35 36@app.post("/predict")37def predict_icd10(input: TextInput):38    for model_name in MODELS:39        try:40            print(f"🔹 Trying remote model: {model_name}")41            result = query_model(model_name, input.text)42            return {43                "model_used": model_name,44                "input": input.text,45                "raw_response": result46            }47        except Exception as e:48            print(f"⚠️ {model_name} failed: {e}")49            continue50    return {"error": "❌ All remote models failed to respond"}51