Update app.py

app.py

@@ -287,7 +287,13 @@ def generate_30(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file.any():  # Check if a latent file is provided
         #sd_image_a = torch.load(latent_file.name)  # Load the latent
-        sd_image_a = latent_file #Image.open(latent_file)
+        #sd_image_a = latent_file #Image.open(latent_file)
+        if not isinstance(latent_file, Image.Image):
+            image = Image.fromarray(latent_file) 
+        image = image.resize((width, height))  # Example resize, adjust as needed
+        image = image.convert("RGB")  # Ensure it's a 3-channel image
+        image = np.array(image)  
+        sd_image_a = torch.from_numpy(image).permute(2, 0, 1) / 127.5 - 1.0  # Normalize
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.txt'
         print("-- using image file --")
@@ -341,7 +347,13 @@ def generate_60(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file.any():  # Check if a latent file is provided
         #sd_image_a = torch.load(latent_file.name)  # Load the latent
-        sd_image_a = latent_file #Image.open(latent_file)
+        #sd_image_a = latent_file #Image.open(latent_file)
+        if not isinstance(latent_file, Image.Image):
+            image = Image.fromarray(latent_file) 
+        image = image.resize((width, height))  # Example resize, adjust as needed
+        image = image.convert("RGB")  # Ensure it's a 3-channel image
+        image = np.array(image)  
+        sd_image_a = torch.from_numpy(image).permute(2, 0, 1) / 127.5 - 1.0  # Normalize
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.txt'
         print("-- using image file --")
@@ -395,7 +407,13 @@ def generate_90(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file.any():  # Check if a latent file is provided
         #sd_image_a = torch.load(latent_file.name)  # Load the latent
-        sd_image_a = latent_file #Image.open(latent_file)
+        #sd_image_a = latent_file #Image.open(latent_file)
+        if not isinstance(latent_file, Image.Image):
+            image = Image.fromarray(latent_file) 
+        image = image.resize((width, height))  # Example resize, adjust as needed
+        image = image.convert("RGB")  # Ensure it's a 3-channel image
+        image = np.array(image)  
+        sd_image_a = torch.from_numpy(image).permute(2, 0, 1) / 127.5 - 1.0  # Normalize
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.txt'
         print("-- using image file --")