TrainAI / requirements.txt
szili2011's picture
Update requirements.txt
1d651bd verified
raw
history blame contribute delete
536 Bytes
# UI Framework
gradio==3.50.2 # Pinning Gradio for stability as well
# Data Handling & Numerical Ops
pandas>=1.3 # pandas is usually pretty stable with forward compatibility
numpy>=1.21
# Core Machine Learning
scikit-learn>=1.1 # Let scikit-learn be somewhat flexible
torch>=1.13
torchvision>=0.14
# ONNX Support for model interoperability
# Pinning these to known compatible versions to resolve the ImportError
skl2onnx==1.16.0
onnx==1.15.0
onnxruntime==1.16.3 # Pinning onnxruntime to a version close to onnx/skl2onnx release times