Update app.py

app.py

@@ -11,8 +11,13 @@ from tqdm import tqdm
 
 from colpali_engine.models import ColQwen2, ColQwen2Processor
 
-print("Install FA2")
-os.system("pip install flash-attn --no-build-isolation")
+
+@spaces.GPU
+def install_fa2():
+    print("Install FA2")
+    os.system("pip install flash-attn --no-build-isolation")
+install_fa2()
+
 
 model = ColQwen2.from_pretrained(
         "manu/colqwen2-v1.0-alpha",