Spaces:
Running
on
Zero
Running
on
Zero
Upload 3 files
Browse files- README.md +10 -6
- requirements.txt +1 -1
README.md
CHANGED
@@ -4,12 +4,15 @@ emoji: 🖼️
|
|
4 |
colorFrom: blue
|
5 |
colorTo: green
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 4.28.3 #
|
8 |
app_file: app.py
|
9 |
-
license: apache-2.0 #
|
10 |
-
#
|
11 |
-
#
|
12 |
-
# hardware: cpu-upgrade
|
|
|
|
|
|
|
13 |
# hf_token: YOUR_HF_TOKEN # Use secrets instead!
|
14 |
---
|
15 |
|
@@ -28,4 +31,5 @@ Model Repository: [celstk/wd-eva02-lora-onnx](https://huggingface.co/celstk/wd-e
|
|
28 |
|
29 |
**Note:**
|
30 |
- This Space uses a model from a **private** repository (`celstk/wd-eva02-lora-onnx`). You might need to duplicate this space and add your Hugging Face token (`HF_TOKEN`) to the Space secrets to allow downloading the model files.
|
31 |
-
- Image pasting behavior might vary across browsers.
|
|
|
|
4 |
colorFrom: blue
|
5 |
colorTo: green
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 4.28.3 # Make sure this matches your gradio version
|
8 |
app_file: app.py
|
9 |
+
license: apache-2.0 # Or your preferred license
|
10 |
+
# Hardware Selection:
|
11 |
+
# For CPU execution (recommended if GPU isn't strictly needed):
|
12 |
+
# hardware: cpu-upgrade
|
13 |
+
# For GPU execution (requires compatible CUDA setup):
|
14 |
+
# hardware: cuda-t4-small
|
15 |
+
# pinned: false # Set to true if you want to pin the hardware
|
16 |
# hf_token: YOUR_HF_TOKEN # Use secrets instead!
|
17 |
---
|
18 |
|
|
|
31 |
|
32 |
**Note:**
|
33 |
- This Space uses a model from a **private** repository (`celstk/wd-eva02-lora-onnx`). You might need to duplicate this space and add your Hugging Face token (`HF_TOKEN`) to the Space secrets to allow downloading the model files.
|
34 |
+
- Image pasting behavior might vary across browsers.
|
35 |
+
- If you require GPU acceleration, uncomment the `hardware: cuda-t4-small` line above and ensure the environment has the necessary CUDA libraries compatible with `onnxruntime-gpu`. The current setup defaults to CPU due to potential CUDA library mismatches in the standard Spaces environment.
|
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
onnxruntime
|
2 |
numpy
|
3 |
Pillow
|
4 |
matplotlib
|
|
|
1 |
+
onnxruntime>=1.16.0 # CPU版
|
2 |
numpy
|
3 |
Pillow
|
4 |
matplotlib
|