Update app.py

app.py

@@ -84,6 +84,30 @@ def find_cuda_directories(search_paths=None):
 #import hidet
 #print(dir(hidet)) 
 #import torch_tensorrt
+import random
+import uuid
+import gradio as gr
+import numpy as np
+from PIL import Image
+
+import diffusers
+from diffusers import AutoencoderKL, StableDiffusionXLPipeline
+from diffusers import EulerAncestralDiscreteScheduler
+from typing import Tuple
+import paramiko
+import datetime
+import cyper
+from image_gen_aux import UpscaleWithModel
+import torch
+#import torch._dynamo
+#torch._dynamo.list_backends()
+import time 
+import gc
+
+import torch.nn.functional as F
+from sageattention import sageattn
+
+
 import pythran
 
 with open("pyt.py", "w") as f:
@@ -116,28 +140,6 @@ os.environ['DISTUTILS_USE_SDK'] = '1'  # Force setuptools (distutils) to use the
 subprocess.run(['pythran', './pyt.py', '-DUSE_XSIMD', '-fopenmp', '-march=native'])
 import pyt
 
-import random
-import uuid
-import gradio as gr
-import numpy as np
-from PIL import Image
-
-import diffusers
-from diffusers import AutoencoderKL, StableDiffusionXLPipeline
-from diffusers import EulerAncestralDiscreteScheduler
-from typing import Tuple
-import paramiko
-import datetime
-import cyper
-from image_gen_aux import UpscaleWithModel
-import torch
-#import torch._dynamo
-#torch._dynamo.list_backends()
-import time 
-import gc
-
-import torch.nn.functional as F
-from sageattention import sageattn
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False