masszhou commited on
Commit
3e30eae
·
1 Parent(s): 7d6b4eb

Add application file

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,4 +1,6 @@
1
  import os
 
 
2
  import gc
3
  import hashlib
4
  import queue
 
1
  import os
2
+ os.system("pip install ./ort_nightly_gpu-1.17.0.dev20240118002-cp310-cp310-manylinux_2_28_x86_64.whl")
3
+ # os.system("pip install ort-nightly-gpu --index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/")
4
  import gc
5
  import hashlib
6
  import queue
requirements.txt CHANGED
@@ -8,4 +8,4 @@ pydantic==2.8.2
8
  fastapi==0.112.2
9
  scipy==1.15.2
10
  numpy==2.2.4
11
- onnxruntime==1.21.0
 
8
  fastapi==0.112.2
9
  scipy==1.15.2
10
  numpy==2.2.4
11
+