clinical-trial-prediction / requirements.txt
archis99's picture
fix: Add torch CPU to requirements
ee2b5f8
raw
history blame contribute delete
444 Bytes
# Core libraries
numpy>=1.26.4,<2.2
pandas==2.2.2
scikit-learn==1.2.2
joblib==1.3.2
# FastAPI and server
fastapi==0.111.0
uvicorn==0.29.0
# Data visualization / plotting
matplotlib==3.9.0
seaborn==0.13.2
# Streamlit
streamlit==1.35.0
# # PyTorch
# torch==2.3.0
# torchvision==0.18.0
# torchaudio==2.3.0
--extra-index-url https://download.pytorch.org/whl/cpu
torch
# Transformers / HuggingFace
transformers==4.43.0
huggingface_hub==0.23.4