Update app.py

app.py

@@ -285,7 +285,7 @@ def generate_30(
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
-        sd_image_a = Image.open(latent_file.name).to(torch.device('cuda'))
+        sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
@@ -338,7 +338,7 @@ def generate_60(
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
-        sd_image_a = Image.open(latent_file.name).to(torch.device('cuda'))
+        sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
@@ -391,7 +391,7 @@ def generate_90(
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
-        sd_image_a = Image.open(latent_file.name).to(torch.device('cuda'))
+        sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")