Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
•
215e62d
1
Parent(s):
0f05ebf
app update
Browse files- app.py +5 -0
- 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
|