diff --git a/app.py b/app.py
index 2a7c788..1368c03 100644
--- a/app.py
+++ b/app.py
@@ -17,12 +17,12 @@
 print('Loading diffusion model ...')
 transformer = FluxTransformer2DModel.from_pretrained(
     "xiaozaa/catvton-flux-alpha", 
-    torch_dtype=device
+    torch_dtype=dtype
 )
 pipe = FluxFillPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     transformer=transformer,
-    torch_dtype=device
+    torch_dtype=dtype
 ).to(device)
 print('Loading Finished!')