PierreHanna commited on
Commit
92791f8
1 Parent(s): 20c4b17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -13,13 +13,15 @@ import json
13
 
14
  embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
15
  embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
16
- # NO GPU
17
- # os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 
 
18
  os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
19
 
20
  # tensorflow.config.threading.set_intra_op_parallelism_threads(8)
21
  # tensorflow.config.threading.set_inter_op_parallelism_threads(8)
22
- print(f"Available GPUs {tensorflow.config.list_physical_devices('GPU')}")
23
  #
24
  python_path = hf_hub_download(
25
  repo_id=os.environ["REPO_ID"],
 
13
 
14
  embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
15
  embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
16
+
17
+ # NO GPU ?
18
+ os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
19
+
20
  os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
21
 
22
  # tensorflow.config.threading.set_intra_op_parallelism_threads(8)
23
  # tensorflow.config.threading.set_inter_op_parallelism_threads(8)
24
+ #print(f"Available GPUs {tensorflow.config.list_physical_devices('GPU')}")
25
  #
26
  python_path = hf_hub_download(
27
  repo_id=os.environ["REPO_ID"],