TB Detection AI: DenseNet-121 Chest X-Ray Diagnostic Assistant

# TB Detection AI Model Inference
import torch
import torchvision.transforms as transforms
from PIL import Image

def predict_tb(image_path, model):
    transform = transforms.Compose([
        transforms.Resize((224, 224)),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
    ])
    
    img = Image.open(image_path).convert('RGB')
    tensor = transform(img).unsqueeze(0)
    
    with torch.no_grad():
        outputs = model(tensor)
        prob = torch.sigmoid(outputs).item()
        
    return {"tb_probability": round(prob, 4), "is_positive": prob > 0.5}