Update app.py

app.py

@@ -195,9 +195,9 @@ def infer(
             width=width,
             height=height,
             latents=sd_image_a,
-            output='latent',
+          #  output='latent',
             generator=generator
-        ).images
+        ).images[0]
     else:
         print('-- generating image --')
         #with torch.no_grad():
@@ -211,17 +211,17 @@ def infer(
             width=width,
             height=height,
          #   latents=None,
-            output='latent',
+          #  output='latent',
             generator=generator,
             max_sequence_length=512
-        ).images
+        ).images[0]
         print('-- got image --')
-    sd35_image = pipe.vae.decode(sd_image / 0.18215).sample
-    sd35_image = sdxl_image.cpu().permute(0, 2, 3, 1).float().detach().numpy()
-    sd35_image = (sdxl_image * 255).round().astype("uint8")
-    image_pil = Image.fromarray(sd35_image[0])
-    sd35_path = f"tst_rv_{seed}.png"
-    image_pil.save(sd35_path,optimize=False,compress_level=0)
+    #sd35_image = pipe.vae.decode(sd_image / 0.18215).sample
+    #sd35_image = sdxl_image.cpu().permute(0, 2, 3, 1).float().detach().numpy()
+    #sd35_image = (sdxl_image * 255).round().astype("uint8")
+    #image_pil = Image.fromarray(sd35_image[0])
+    sd35_path = f"sd35_{seed}.png"
+    sd_image.save(sd35_path,optimize=False,compress_level=0)
     upload_to_ftp(sd35_path)
     
         # Convert the generated image to a tensor