Zengyf-CVer commited on
Commit
215e62d
1 Parent(s): 0f05ebf

app update

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. requirements.txt +2 -3
app.py CHANGED
@@ -4,6 +4,11 @@
4
  # 功能描述:多选,多项模型转换和打包下载
5
 
6
  import os
 
 
 
 
 
7
  import shutil
8
  import time
9
  import zipfile
 
4
  # 功能描述:多选,多项模型转换和打包下载
5
 
6
  import os
7
+
8
+ os.system("pip install pip -U")
9
+ os.system("pip install nvidia-pyindex")
10
+ os.system("pip install nvidia-tensorrt")
11
+
12
  import shutil
13
  import time
14
  import zipfile
requirements.txt CHANGED
@@ -3,7 +3,6 @@
3
  # pip 22.1.2
4
 
5
  # Base ----------------------------------------
6
- pip==22.1.2
7
  matplotlib==3.5.2
8
  numpy==1.19.5
9
  opencv-python==4.5.5.64
@@ -34,8 +33,8 @@ coremltools==5.2.0 # CoreML export
34
  onnx==1.12.0 # ONNX export
35
  onnx-simplifier==0.4.0 # ONNX simplifier
36
  onnxruntime-gpu==1.11.1
37
- nvidia-pyindex==1.0.9
38
- nvidia-tensorrt==8.4.1.5
39
  scikit-learn==0.24.2 # CoreML quantization
40
  tensorflow==2.4.1 # TFLite export
41
  tensorflowjs==3.18.0 # TF.js export
 
3
  # pip 22.1.2
4
 
5
  # Base ----------------------------------------
 
6
  matplotlib==3.5.2
7
  numpy==1.19.5
8
  opencv-python==4.5.5.64
 
33
  onnx==1.12.0 # ONNX export
34
  onnx-simplifier==0.4.0 # ONNX simplifier
35
  onnxruntime-gpu==1.11.1
36
+ # nvidia-pyindex==1.0.9
37
+ # nvidia-tensorrt==8.4.1.5
38
  scikit-learn==0.24.2 # CoreML quantization
39
  tensorflow==2.4.1 # TFLite export
40
  tensorflowjs==3.18.0 # TF.js export