Spaces:
Running
on
Zero
Running
on
Zero
Upload requirements.txt
Browse files- requirements.txt +2 -3
requirements.txt
CHANGED
@@ -1,11 +1,10 @@
|
|
1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
2 |
torch
|
3 |
torchvision
|
4 |
torchaudio
|
5 |
-
onnxruntime-gpu
|
6 |
numpy
|
7 |
Pillow
|
8 |
matplotlib
|
9 |
requests
|
10 |
-
gradio==4.44.1
|
11 |
huggingface_hub
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
torch
|
3 |
torchvision
|
4 |
torchaudio
|
5 |
+
onnxruntime-gpu==1.18.0 # または onnxruntime>=1.16.0 (CPUのみの場合)
|
6 |
numpy
|
7 |
Pillow
|
8 |
matplotlib
|
9 |
requests
|
|
|
10 |
huggingface_hub
|