Update app.py

app.py

@@ -2,6 +2,17 @@ import subprocess
 subprocess.run(['sh', './spaces.sh'])
 import spaces
 
+@spaces.GPU(required=True)
+def install_dependencies():
+    subprocess.run(['sh', './flashattn.sh'])
+    # Install the UltraHDR library
+    print("Installing pillow-ultrahdr...")
+    subprocess.run(['pip', 'install', 'pillow-ultrahdr'])
+    print("✅ pillow-ultrahdr installed.")
+
+# Install all dependencies
+install_dependencies()
+
 import os
 
 os.environ['PYTORCH_NVML_BASED_CUDA_CHECK'] = '1'
@@ -44,16 +55,6 @@ from google.cloud import storage
 
 import torch
 
-@spaces.GPU(required=True)
-def install_dependencies():
-    subprocess.run(['sh', './flashattn.sh'])
-    # Install the UltraHDR library
-    print("Installing pillow-ultrahdr...")
-    subprocess.run(['pip', 'install', 'pillow-ultrahdr'])
-    print("✅ pillow-ultrahdr installed.")
-
-# Install all dependencies
-# install_dependencies()
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False