PierreHanna commited on
Commit
5f755a3
·
verified ·
1 Parent(s): ba774ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ embed_html2 = '" title="YouTube video player" frameborder="0" allow="acceleromet
15
  os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
16
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
17
 
18
- tf.config.threading.set_intra_op_parallelism_threads(6)
19
- tf.config.threading.set_inter_op_parallelism_threads(6)
20
 
21
  #
22
  python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],
 
15
  os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
16
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
17
 
18
+ tensorflow.config.threading.set_intra_op_parallelism_threads(6)
19
+ tensorflow.config.threading.set_inter_op_parallelism_threads(6)
20
 
21
  #
22
  python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],