Update requirements.txt
Browse files- requirements.txt +10 -9
requirements.txt
CHANGED
|
@@ -1,16 +1,17 @@
|
|
| 1 |
# UI Framework
|
| 2 |
-
gradio
|
| 3 |
|
| 4 |
# Data Handling & Numerical Ops
|
| 5 |
-
pandas>=1.3
|
| 6 |
-
numpy>=1.21
|
| 7 |
|
| 8 |
# Core Machine Learning
|
| 9 |
-
scikit-learn>=1.1 #
|
| 10 |
-
torch>=1.13
|
| 11 |
-
torchvision>=0.14
|
| 12 |
|
| 13 |
# ONNX Support for model interoperability
|
| 14 |
-
|
| 15 |
-
|
| 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
|