kjerk commited on
Commit
a46979d
·
1 Parent(s): b4533b7

Switch to cuda enabled for testing

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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 = bool(os.getenv("ALLOW_CUDA", "false").lower())
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