Update app.py

app.py

@@ -4,18 +4,16 @@ import spaces
 import os
 import numpy as np
 from PIL import Image
-from huggingface_hub import hf_hub_download
-from safetensors.torch import load_file
 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
+