Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|