Update app.py

app.py

@@ -59,7 +59,7 @@ def load_models():
     )
 
     unet.requires_grad_(False)
-    unet.to(device, dtype=weight_dtype)
+    unet.to(device.value, dtype=weight_dtype)
     vae.requires_grad_(False)
 
     text_encoder.requires_grad_(False)