Update app.py

app.py

@@ -6,6 +6,12 @@
 # copies of the Software, and to permit persons to whom the Software is
 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'
@@ -107,11 +113,6 @@ ip_ckpt = os.path.join(local_folder2, "ip-adapter_sdxl_vit-h.bin") # Correct pat
 
 upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
 
-@spaces.GPU(required=True)
-def install_flashattn():
-    subprocess.run(['sh', './flashattn.sh'])
-    
-#install_flashattn()
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     if style_name in styles: