Update app.py

app.py

@@ -1,13 +1,16 @@
+
+import os
+os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
+os.environ["OMP_NUM_THREADS"] = "1"
+os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
+
+import os.path as osp
 import torch
 import cv2
 import numpy as np
-import os
-import os.path as osp
 import time
 import gradio as gr
-os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
-os.environ["OMP_NUM_THREADS"] = "1"
-os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
+
 
 from models.TextEnhancement import MARCONetPlus
 from utils.utils_image import imread_uint, uint2tensor4, tensor2uint