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
|