Upload 2 files

README.md

@@ -6,13 +6,10 @@ colorTo: green
 sdk: gradio
 sdk_version: 4.28.3 # Make sure this matches your gradio version
 app_file: app.py
-license: apache-2.0 # Or your preferred license
-# Hardware Selection:
-# For CPU execution (recommended if GPU isn't strictly needed):
-# hardware: cpu-upgrade 
-# For GPU execution (requires compatible CUDA setup):
-# hardware: cuda-t4-small 
-# pinned: false # Set to true if you want to pin the hardware
+license: apache-2.0 # または適切なライセンス
+# Pinned Hardware: T4 small (GPU) or CPU upgrade (CPU)
+# pinned: false # 必要に応じてTrueに
+# hardware: cpu-upgrade # or cuda-t4-small
 # hf_token: YOUR_HF_TOKEN # Use secrets instead!
 ---
 

requirements.txt

@@ -1,7 +1,11 @@
-onnxruntime>=1.16.0 # CPU版
+--extra-index-url https://download.pytorch.org/whl/cu126
+torch
+torchvision 
+torchaudio
+onnxruntime-gpu>=1.16.0 # または onnxruntime>=1.16.0 (CPUのみの場合)
 numpy
 Pillow
 matplotlib
 requests
-gradio
+gradio=4.44.1
 huggingface_hub