ChestViT-Explainable-XRay-AI / requirements.txt
sandy45's picture
fix: add missing dependencies (grad-cam, datasets, faiss-cpu, sentence-transformers)
d781259
Raw
History Blame Contribute Delete
2.33 kB
# ─── Core ML ────────────────────────────────────────────────────────────────
torch>=2.1.0
torchvision>=0.16.0
torchaudio>=2.1.0
# ─── HuggingFace ─────────────────────────────────────────────────────────────
transformers>=4.37.0
accelerate>=0.26.0
# ─── Image Processing ────────────────────────────────────────────────────────
opencv-python>=4.9.0.80
albumentations>=1.3.1
Pillow>=10.2.0
grad-cam>=1.5.0
# ─── Hugging Face Datasets ───────────────────────────────────────────────────
datasets>=2.16.0
# ─── Data Science ────────────────────────────────────────────────────────────
numpy>=1.26.0
pandas>=2.1.0
scikit-learn>=1.4.0
matplotlib>=3.8.0
seaborn>=0.13.0
faiss-cpu>=1.7.4
sentence-transformers>=2.3.1
# ─── Experiment Tracking ─────────────────────────────────────────────────────
mlflow>=2.10.0
# ─── Demo UI ─────────────────────────────────────────────────────────────────
gradio>=5.33.0
audioop-lts # Python 3.13 removed audioop; audioop-lts backports it for pydub/gradio
# ─── Dataset Download ────────────────────────────────────────────────────────
kaggle>=1.6.0
# ─── Utilities ───────────────────────────────────────────────────────────────
tqdm>=4.66.0
pyyaml>=6.0.1
rich>=13.7.0