Update app.py

app.py

@@ -49,6 +49,7 @@ def generate(image: Image.Image, edit_prompt: str):
     # dit.to(device, dtype=dtype)
 
     sampler = XFluxSampler(
+        device = device
         ip_loaded=False,
         spatial_condition=False,
         clip_image_processor=None,