Update app.py

app.py

@@ -81,15 +81,7 @@ pipe = Flux2Pipeline.from_pretrained(
 pipe.to(device)
 
 # Pull pre-compiled Flux2 Transformer blocks from HF hub
-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")
+spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa3")
 
 def image_to_data_uri(img):
     buffered = io.BytesIO()