csxmli commited on
Commit
9eba4f2
·
verified ·
1 Parent(s): 27a4300

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os.path as osp
6
  import time
7
  import gradio as gr
8
  os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
9
-
10
  os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
11
 
12
  from models.TextEnhancement import MARCONetPlus
 
6
  import time
7
  import gradio as gr
8
  os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
9
+ os.environ["OMP_NUM_THREADS"] = "1"
10
  os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
11
 
12
  from models.TextEnhancement import MARCONetPlus