Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import subprocess
 import sys
 import io
@@ -76,6 +77,9 @@ pipe = Flux2Pipeline.from_pretrained(
 )
 pipe.to(device)
 
+# Pull pre-compiled Flux2 Transformer blocks from HF hub
+spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2")
+
 def image_to_data_uri(img):
     buffered = io.BytesIO()
     img.save(buffered, format="PNG")