Update app.py

app.py

@@ -7,14 +7,13 @@ from PIL import Image
 from omegaconf import OmegaConf
 
 from image_datasets.dataset import image_resize
-from src.flux.xflux_pipeline import XFluxSampler
 args = OmegaConf.load("inference_configs/inference.yaml")
 device = torch.device("cuda")
 dtype = torch.bfloat16
 @spaces.GPU
 def generate(image: Image.Image, edit_prompt: str):
     
-
+    from src.flux.xflux_pipeline import XFluxSampler
     
 
     sampler = XFluxSampler(