MDS_demonstrator / requirements.txt
AMontiB
update ALL
6a489d1
# ML / DL Frameworks
torch==2.4.0
torchvision==0.19
transformers
# ML/DL Helpers
einops==0.8.0
# Data Science
# Using headless opencv to prevent libGL errors in Docker/Spaces
opencv-python-headless==4.10.0.84
pandas==2.2.3
scikit-image==0.22.0
scikit-learn==1.5.2
numpy<2.0
matplotlib
# MLOps & Infrastructure
wandb==0.16.6
# Utilities
PyYAML==6.0.2
tqdm==4.67.1
albumentations
ftfy
open-clip-torch
# 1. Use latest Gradio
gradio==5.49.1
# 2. Unpin FastAPI and Pydantic to allow latest versions
# fastapi
# pydantic