Update app.py

app.py

@@ -36,6 +36,8 @@ models_path = snapshot_download(repo_id="Snapchat/w2w")
 
 
 
+
+device="cuda"
 pretrained_model_name_or_path = "stablediffusionapi/realistic-vision-v51" 
 revision = None
 weight_dtype = torch.bfloat16
@@ -66,8 +68,6 @@ text_encoder.to(device, dtype=weight_dtype)
 print("")
 
     
-
-device="cuda"
 mean = torch.load(f"{models_path}/files/mean.pt", map_location=torch.device('cpu')).bfloat16().to(device)
 std = torch.load(f"{models_path}/files/std.pt", map_location=torch.device('cpu')).bfloat16().to(device)
 v = torch.load(f"{models_path}/files/V.pt", map_location=torch.device('cpu')).bfloat16().to(device)