Update app.py

app.py

@@ -81,7 +81,15 @@ 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", variant="fa3")
+try:
+    spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa3")
+except FileNotFoundError:
+    print("FA3 variant not available, trying FA2...")
+    try:
+        spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa2")
+    except FileNotFoundError:
+        print("FA2 variant not available, using default...")
+        spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2")
 
 def image_to_data_uri(img):
     buffered = io.BytesIO()