cheng-hust commited on
Commit
abe6a95
·
verified ·
1 Parent(s): 111ae13

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -1
requirements.txt CHANGED
@@ -1,5 +1,11 @@
1
- torch
2
  gradio
 
 
 
 
 
 
 
3
  pillow
4
  opencv-python
5
  pathlib
 
 
1
  gradio
2
+ torch==2.0.1
3
+ torchvision==0.15.2
4
+ onnx==1.14.0
5
+ onnxruntime==1.15.1
6
+ pycocotools
7
+ PyYAML
8
+ scipy
9
  pillow
10
  opencv-python
11
  pathlib