Update app.py

app.py

@@ -82,7 +82,7 @@
 import gradio as gr
 import torch
 from diffusers import FluxPipeline
-from torchao import swap_conv2d_1x1_to_linear, apply_dynamic_quant
+from torchao.quantization import swap_conv2d_1x1_to_linear, apply_dynamic_quant
 
 # Step 1: Enable PyTorch 2-specific optimizations
 torch._inductor.config.conv_1x1_as_mm = True