Update app.py

app.py

@@ -36,9 +36,14 @@ scheduler = DDIMScheduler.from_pretrained(
     steps_offset=1,
     token=os.getenv("hf_token")
 )
-
+flx = FluxPipeline.from_pretrained(
+    base,
+    torch_dtype=dtype,
+    token=os.getenv("hf_token")
+).to(device)
 pipe = AnimateDiffSDXLPipeline.from_pretrained(
     base,
+    unet=flx.components.unet,
     motion_adapter=adapter,
     scheduler=scheduler,
     torch_dtype=dtype,