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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -15,6 +15,9 @@ 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
  #
19
  python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],
20
  use_auth_token=os.environ['TOKEN'])
 
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'],
23
  use_auth_token=os.environ['TOKEN'])