Update app.py

app.py

@@ -1,6 +1,14 @@
 import subprocess
 subprocess.run(['sh', './spaces.sh'])
 
+import spaces
+
+@spaces.GPU(required=True)
+def install_flashattn():
+    subprocess.run(['sh', './flashattn.sh'])
+
+install_flashattn()
+
 import os
 
 os.environ['PYTORCH_NVML_BASED_CUDA_CHECK'] = '1'
@@ -27,7 +35,6 @@ torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 
-import spaces
 import gradio as gr
 import numpy as np
 import random
@@ -39,11 +46,6 @@ import io
 from google.oauth2 import service_account
 from google.cloud import storage
 
-@spaces.GPU(required=True)
-def install_flashattn():
-    subprocess.run(['sh', './flashattn.sh'])
-
-#install_flashattn()
 
 from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL
 from PIL import Image