Spaces:
Sleeping
Sleeping
Switch to cuda enabled for testing
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -67,7 +67,7 @@ def bootstrap_model():
|
|
| 67 |
huggingface_repo = os.getenv("HF_REPO_NAME", "")
|
| 68 |
file_path = os.getenv("HF_FILE_PATH", "")
|
| 69 |
access_token = os.getenv("HF_TOKEN", "")
|
| 70 |
-
allow_cuda =
|
| 71 |
|
| 72 |
model_file_path = fetch_model_to_cache(huggingface_repo, file_path, access_token)
|
| 73 |
|
|
|
|
| 67 |
huggingface_repo = os.getenv("HF_REPO_NAME", "")
|
| 68 |
file_path = os.getenv("HF_FILE_PATH", "")
|
| 69 |
access_token = os.getenv("HF_TOKEN", "")
|
| 70 |
+
allow_cuda = os.getenv("ALLOW_CUDA", "false").lower() in {'true', 'yes', '1', 'y'}
|
| 71 |
|
| 72 |
model_file_path = fetch_model_to_cache(huggingface_repo, file_path, access_token)
|
| 73 |
|
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
blended_tiling==0.0.1.dev7
|
| 2 |
huggingface-hub==0.27.1
|
| 3 |
-
onnxruntime==1.20.1
|
|
|
|
| 4 |
pillow==11.0.0
|
| 5 |
streamlit==1.41.1
|
| 6 |
streamlit-image-comparison==0.0.4
|
|
|
|
| 1 |
blended_tiling==0.0.1.dev7
|
| 2 |
huggingface-hub==0.27.1
|
| 3 |
+
#onnxruntime==1.20.1
|
| 4 |
+
onnxruntime-gpu==1.20.1
|
| 5 |
pillow==11.0.0
|
| 6 |
streamlit==1.41.1
|
| 7 |
streamlit-image-comparison==0.0.4
|