cella110n commited on
Commit
8244653
·
verified ·
1 Parent(s): 0885640

Upload 3 files

Browse files
Files changed (2) hide show
  1. README.md +10 -6
  2. 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 # requirements.txt と合わせるか確認
8
  app_file: app.py
9
- license: apache-2.0 # または適切なライセンス
10
- # Pinned Hardware: T4 small (GPU) or CPU upgrade (CPU)
11
- # pinned: false # 必要に応じてTrueに
12
- # hardware: cpu-upgrade # or cuda-t4-small
 
 
 
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-gpu>=1.16.0 # または onnxruntime>=1.16.0 (CPUのみの場合)
2
  numpy
3
  Pillow
4
  matplotlib
 
1
+ onnxruntime>=1.16.0 # CPU
2
  numpy
3
  Pillow
4
  matplotlib