This repository provides two trained deep learning models for classifying X-ray images from the TAIX-Ray dataset:
-
Binary Classification Model - Classifies X-ray images into two categories (normal vs. abnormal).
-
Ordinal Classification Model - Predicts severity levels based on ordinal categories.
Model definitions, training, and evaluation code are available at
https://github.com/mueller-franzes/TAIX-Ray
1from huggingface_hub import hf_hub_download
2
3
4# Download the checkpoint file from Hugging Face Hub
5file_path = hf_hub_download(
6 repo_id="TLAIM/TAIX-Ray",
7 filename="binary.ckpt", # binary.ckpt or ordinal.ckpt
8)
9
10# Check if the file has been correctly downloaded
11print(f"Checkpoint downloaded to: {file_path}")