szili2011 commited on
Commit
1d651bd
·
verified ·
1 Parent(s): b1c6c9b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -9
requirements.txt CHANGED
@@ -1,16 +1,17 @@
1
  # UI Framework
2
- gradio>=3.50.2
3
 
4
  # Data Handling & Numerical Ops
5
- pandas>=1.3
6
- numpy>=1.21 # scikit-learn and torch often have numpy bounds
7
 
8
  # Core Machine Learning
9
- scikit-learn>=1.1 # For good compatibility with modern features
10
- torch>=1.13 # Main deep learning library
11
- torchvision>=0.14 # For image utilities, usually aligned with torch
12
 
13
  # ONNX Support for model interoperability
14
- skl2onnx>=1.16
15
- onnx>=1.15
16
- onnxruntime>=1.16
 
 
1
  # UI Framework
2
+ gradio==3.50.2 # Pinning Gradio for stability as well
3
 
4
  # Data Handling & Numerical Ops
5
+ pandas>=1.3 # pandas is usually pretty stable with forward compatibility
6
+ numpy>=1.21
7
 
8
  # Core Machine Learning
9
+ scikit-learn>=1.1 # Let scikit-learn be somewhat flexible
10
+ torch>=1.13
11
+ torchvision>=0.14
12
 
13
  # ONNX Support for model interoperability
14
+ # Pinning these to known compatible versions to resolve the ImportError
15
+ skl2onnx==1.16.0
16
+ onnx==1.15.0
17
+ onnxruntime==1.16.3 # Pinning onnxruntime to a version close to onnx/skl2onnx release times